題目描述:
聖誕老人共有MM個餅乾,準備全部分給N個孩子。
每個孩子有一個貪婪度,第 i 個孩子的貪婪度為 g[i]。
如果有 a[i]a[i] 個孩子拿到的餅乾數比第 ii 個孩子多,那麼第 ii 個孩子會產生 g[i]×a[i]的怨氣。
給定N、M和序列g,聖誕老人請你幫他安排一種分配方式,使得每個孩子至少分到一塊餅乾,並且所有孩子的怨氣總和最小。
輸入格式
第一行包含兩個整數N,M。第二行包含N個整數表示g1~gN。
輸出格式
第一行一個整數表示最小怨氣總和。第二行N個空格隔開的整數表示每個孩子分到的餅乾數,若有多種方案,輸出任意一種均可。
資料範圍
1≤N≤30,N≤M≤5000,1≤gi≤10^7
輸入樣例:
3 20
1 2 3
輸出樣例:
2
2 9 9
解題思路:
完整程式碼:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
inline int read()
{
char c=getchar();int f=1,s=0;
while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){s=s*10+c-'0';c=getchar();}
return f*s;
}
int f[31][5010];
int n,m;
struct child
{
int d,ip;
}g[31];
inline bool cmp(child n1,child n2){return n1.d>n2.d;}
inline int mymin(int x,int y){return x<y?x:y;}
int cumu[31];
struct state
{
int nd,t;
}a[31][5010];
int ans[31];
int main()
{
n=read();m=read();
for(int i=1;i<=n;i++)
g[i].d=read(),g[i].ip=i;
sort(g+1,g+n+1,cmp);
memset(f,63,sizeof(f));
for(int i=1;i<=n;i++)cumu[i]=cumu[i-1]+g[i].d,f[i][0]=0,a[i][0].nd=0,a[i][0].t=i;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=m;j++)
{
if(f[i][j-i]<f[i][j])
{
f[i][j]=f[i][j-i];
a[i][j].nd=a[i][j-i].nd+1;
a[i][j].t=a[i][j-i].t;
}
for(int k=1;k<=i;k++)
{
if(f[i-k][j-k]+(i-k)*(cumu[i]-cumu[i-k])<f[i][j])
{
f[i][j]=f[i-k][j-k]+(i-k)*(cumu[i]-cumu[i-k]);
a[i][j].nd=1;a[i][j].t=k;
}
}
}
}
printf("%d\n",f[n][m]);
int now=n,sum=m;
while(now)
{
int dd=a[now][sum].nd;
int tt=a[now][sum].t;
ans[g[now].ip]=dd;
for(int i=now;i>=now-tt+1;i--)
{
ans[g[i].ip]=dd;
}
now-=tt;
sum-=dd*tt;
}
for(int i=1;i<=n;i++)printf("%d ",ans[i]);
return 0;
}