[CH5105]Cookies

题面描述

传送门

思路

这么一道无序的题目，怎么弄出来的DP！

经过$lyd$的玄学引导，详见蓝书。

可以尝试将$g_i$从大到小排序，因为贪婪值较大的，拥有的饼干越多，对答案的贡献就越小。

排序之后，

第$i$个孩子的饼干只有两种情况:

1. 饼干数小于第$i-1$个孩子拥有的饼干数，即$a_i=i-1$
2. 饼干数等于第$i-1$个孩子拥有的饼干数，此时我们要统计在$i$之前有多少个孩子饼干数与$i$相同。

发现很难维护状态，

经过引导，

1. 若第$i$个孩子获得的饼干数$>1$，则每一个人少拿一块饼干，效果是等价的。
2. 若饼干数$=1$，就可以上文第二种情况操作，通过枚举来统计。

那么方程就显而易见了，

$F_{i,j}$表示前$i$个人共有$j$块饼干。

$$F_{i,j}=\min \left\{\begin{array}{c}{F}_{i,j-1}\\ \underset{0\le k<i}{\min }\left\{\begin{array}{c}{F}_{k,j-(i-k)}+k\ast \sum _{p=k+1}^i{g}_p\end{array}\right\}\end{array}\right\}$$

$lyd$太强了！

对了，还要记录饼干数，用两个前继分别记录$i,j$就可以了。

记得对应操作

对应第一种情况

if(pi[n][m]==n)
{
	for(int i=1;i<=n;i++)
		++ans[a[i].id];
}

对应第二种情况

else
{
	for(int i=pi[n][m]+1;i<=n;i++)
		ans[a[i].id]=1;
}

AC code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#define gc getchar()
#define ll long long
using namespace std;
const int M=5e3+10;
struct node{int g,id;}a[31];
bool cmp(node a,node b){return a.g>b.g;}
ll f[31][M],sum[31];int g[31];
int pi[31][M],pj[31][M],ans[31];
void calc(int n,int m)
{
	if(!n)return ;
	calc(pi[n][m],pj[n][m]);
	if(pi[n][m]==n)
	{
		for(int i=1;i<=n;i++)
			++ans[a[i].id];
	}
	else
	{
		for(int i=pi[n][m]+1;i<=n;i++)
			ans[a[i].id]=1;
	}
}
int main()
{
	int n,m;scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){scanf("%d",&a[i].g);a[i].id=i;}
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[i].g;
	memset(f,0x3f,sizeof(f));f[0][0]=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(j-i>=0)
			{
				f[i][j]=f[i][j-i];
				pi[i][j]=i;
				pj[i][j]=j-i;
			}
			int p;
			for(int k=0;k<i;k++)
				if(j-(i-k)>=0&&f[k][p=(j-(i-k))]+k*(sum[i]-sum[k])<f[i][j])
				{
					f[i][j]=f[k][p]+k*(sum[i]-sum[k]);
					pi[i][j]=k;
					pj[i][j]=p;
				}
		}
	}
	printf("%lld\n",f[n][m]);
	calc(n,m);
	for(int i=1;i<=n;i++)printf("%d ",ans[i]);
	puts("");
	return 0;
}