题意:
有一家糖果店,每天可以生产m颗糖果,你在接下来n天会去买糖果,你每天必须吃一颗糖果,你买的糖果可以保存以后吃,你每天买糖果还需要额外花当天买的糖果个数的平方的钱,求这n天你最少花多少钱?
思路:
dp
dp[i][j]表示前i天买j颗糖果的最少花费。
sum[i][o]表示第i天买o颗糖果的糖果花费。
前i天买的糖果为第i天买的糖果加上前i-1买的糖果,注意每天必须吃一颗糖果(既i<=j)。
状态转移方程:
dp[i][j]=min(dp[i-1][j-o]+o*o+sum[i][o]) (j>=i&&j-o>=i-1&&o>=0&&o<=m);(o为第i天买的糖果数)
代码:
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int inf=1000000007;
ll d[305][305];
ll sum[305][305], dp[305][305];
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("%lld",&d[i][j]);
}
sort(d[i]+1,d[i]+m+1);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
sum[i][j]=sum[i][j-1]+d[i][j];
}
}
fill(dp[0],dp[0]+305*305,inf);
dp[0][0]=0;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
dp[i][j]=dp[i-1][j];
for(int o=1;o<=m;o++)
{
if(j>=o&&j-o>=i-1)
dp[i][j]=min(dp[i][j],dp[i-1][j-o]+sum[i][o]+o*o);
}
}
}
cout << dp[n][n] << endl;
return 0;
}

京公网安备 11010502036488号