思路:dp[i][j]表示第i天有j个糖果的最小花费。过%60数据的可能是RE,数组开大一点会MLE,所以需要优化对于每一天买的糖果不需要超过n个糖果。记录前缀和,然后用类似背包方法从第i-1天递推第i天。
#include<iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 305;
int dp[MAXN][MAXN];
int a[MAXN][MAXN];
int main()
{
int n, m;
scanf("%d %d", &n, &m);
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j)
scanf("%d",&a[i][j]);
for(int i=1;i<=n;++i)
sort(*(a+i)+1,*(a+i)+m+1);
for(int i=1;i<=n;++i)
{
for(int j=1;j<=m;++j)
a[i][j]=a[i][j-1]+a[i][j];
}
int k=2;
memset(dp,0x3f,sizeof(dp));
for(int i=1;i<=m;++i)
dp[1][i]=a[1][i]+i*i;
for(int i=2;i<=n;++i)
{
for(int j=i;j<=k*m;++j)
{
dp[i][j]=min(dp[i-1][j],dp[i][j]);
for(int c=1;c<=m;++c)
{
if(j-c>=i-1)
dp[i][j]=min(dp[i][j],dp[i-1][j-c]+c*c+a[i][c]);
}
}
k++;
}
printf("%d\n", dp[n][n]);
return 0;
}