前言:基本的算法思路还是先看数据范围,接着看能不能用动态规划来做,刚刚好这个题目可以套用前---选---
cpp
#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;
}