解析:
因为题目要求价格最低,所以我们需要给m个糖果从小到大排序,然后再计算出每一天的m个糖果的价格前缀和
设f[i][j]:表示第i天,买了j个糖果
状态转移方程:f[i][j]=min(f[i][j],f[i-1][k]+sum[i][j-k]+(j-k)*(j-k));
第一维(第一重循环): 枚举第i天
第二维(第二重循环): 枚举第i天选的糖果个数(这里没判断每天选的糖果个数不能超过m个,因为上面加了一个初始化)。第i天选的糖果个数起始一定是i,因为前i天至少选了i个。
第三重循环: 枚举i-1天选了糖果的个数
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=1e9;
const int N=550;
int a[N][N];
ll sum[N][N];
ll f[N][N],n,m;
int main()
{
scanf("%d %d",&n,&m);
for(int i=0;i<=n;i++)//此处初始化会抵消选糖果数超过m个的情况
{
for(int j=1;j<=400;j++) f[i][j]=sum[i][j]=INF;
}
for(int i=1;i<=n;i++)//处理前缀和
{
for(int j=1;j<=m;j++) cin>>a[i][j];
sort(a[i]+1,a[i]+1+m);
for(int j=1;j<=m;j++) sum[i][j]=sum[i][j-1]+a[i][j];
}
//初始化赋值第一天选m个糖果的情况
for(int i=1;i<=m;i++) f[1][i]=sum[1][i]+i*i;
for(int i=2;i<=n;i++)
{
for(int j=i;j<=n;j++)//j=i,因为枚举第i天选的,因为钱i-1天已经至少选了i-1个,所以j==i
{
for(int k=i-1;k<=j;k++)//k=i-1 因为枚举第i天正要选的,所以已经选好了i-1个
f[i][j]=min(f[i][j],f[i-1][k]+sum[i][j-k]+(j-k)*(j-k));
}
}
cout<<f[n][n]<<endl;
}
解析:
不用初始化限制只能选m个
在循环里加限制
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=1e9;
const int N=550;
int a[N][N];
ll sum[N][N];
ll f[N][N],n,m;
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++) cin>>a[i][j];
sort(a[i]+1,a[i]+1+m);
for(int j=1;j<=m;j++) sum[i][j]=sum[i][j-1]+a[i][j];
}
memset(f,0x3f,sizeof f);
// for(int i=1;i<=m;i++) f[1][i]=sum[1][i]+i*i;
f[0][0]=0;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
for(int k=0;k<=j&&k<=m;k++)
{
// if(f[i-1][j-k]!=0x3f3f3f3f)
//{
f[i][j]=min(f[i][j],f[i-1][j-k]+sum[i][k]+k*k);
// }
}
}
}
cout<<f[n][n]<<endl;
}