前言:基本的算法思路还是先看数据范围,接着看能不能用动态规划来做,刚刚好这个题目可以套用前—选—
题目地址
#include<bits/stdc++.h>
using namespace std;#define int long long
int n,m;
const int N = (int)305;
int dp[N][N]; // 前 i 天中选取 j 个糖果的最小费用
int a[N][N];// 考虑用
signed main(){cin >> n >> m;for(int i = 1;i<=n;i++){vector<int> b; for(int j=1;j<=m;j++){int x; cin >> x;b.push_back(x);}sort(b.begin(),b.end());for(int j=1;j<=m;j++) a[i][j] = b[j-1];}for(int i=1;i<=n;i++){for(int j=1;j<=n;j++){dp[i][j] = 0x3ffffff;}}int now = 0;for(int i=1;i<=m;i++){now += a[1][i];dp[1][i] = now + i*i;}for(int i=2;i<=n;i++){int now = 0;for(int j=1;j<=m;j++){now += a[i][j];// 下面类似背包问题for(int k=i-1;k<=n&&(j+k<=n);k++){dp[i][j+k] = min({dp[i][j+k],dp[i-1][k]+now+j*j,dp[i-1][j+k]});}}}cout << dp[n][n];system("pause");return 0;
}