每天至少要吃一颗糖,一天买k个糖的花费为糖的钱加上k*k,可储存,问最少需要多少钱.首先要想到是dp,我也没想到,惭愧啊,然后就是dp的注意事项了,一共有n天那么最少花费一定是n天吃了n个糖。由这两个状态可以得到dp方程的形式为这样dp[i][j]到第i天一共买了j个糖(i<=j)因为每天要吃一个;
#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<string>
#include<vector>
#include<queue>
#include<algorithm>
#include<deque>
#include<map>
#include<stdlib.h>
#include<set>
#include<iomanip>
#include<stack>
#define ll long long
#define ms(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x & -x
#define fi first
#define ull unsigned long long
#define se second
#define endl "\n"
#define bug cout<<"----acac----"<<endl
#define IOS ios::sync_with_stdio(false), cin.tie(0),cout.tie(0)
using namespace std;
const int maxn =3e5 + 5;
const int maxm = 1.5e5+50;
const double eps = 1e-7;
const double inf = 0x3f3f3f3f;
const double lnf = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;
const int p=1e7+233;
const double pi=3.141592653589;
ll dp[305][305];
ll a[305][305];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
ms(dp,inf);
dp[0][0]=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
scanf("%lld",&a[i][j]);
}
}
for(int i=1;i<=n;i++)
{
sort(a[i]+1,a[i]+1+m);//排序,贪心
for(int j=1;j<=m;j++)
{
a[i][j]+=a[i][j-1];//前缀和
}
}
for(int i=1;i<=n;i++)
{
for(int j=i;j<=min(n,(i)*m);j++)//一定要注意边界,i这个状态的范围
{
for(int k=i-1;k<=min(n,min(j,(i-1)*m));k++)//同上,i-1这个状态的范围
{
dp[i][j]=min(dp[i][j],dp[i-1][k]+a[i][j-k]+(j-k)*(j-k));
}
}
}
printf("%lld\n",dp[n][n]);
return 0;
}