AcWing 277. 饼干 （线性DP，划分）

AcWing 277. 饼干

分析：

• 对于所有的分配方案，有些显然是不需要的，所以先预处理一下

• 要使怨气总和最小，将怨气值从大到小排序，可以想到，怨气值最大分到的饼干最多

  即，分配的饼干数是单调下降的，考虑状态转移

• 设 $f[i][j]$ 表示将j个饼干分给前i个小朋友，且分配的饼干数单调下降的方案数

f [ i ] [ j ] = { f [ i ] [ j − i ] ( 初 值 ， 从 之 前 的 状 态 继 承 ， 即 ， f [ i ] [ j ] = = f [ i ] [ j − i ] ) m i n { f [ i − k ] [ j − k ] + ( s [ i ] − s [ i − k ] ) ( i − k ) } 就 把 i 个 人 分 成 了 两 部 分 ， 一 部 分 人 分 到 大 于 1 个 饼 干 ， 一 部 分 分 到 1 个 饼 干 ; k 表 示 有 k 个 人 分 到 了 一 个 饼 干 ; ( s [ i ] − s [ i − k ] ) 表 示 这 k 个 人 的 怒 气 值 ; ( i − k ) 表 示 有 ( i − k ) 个 人 分 到 大 于 1 个 饼 干 . f[i][j]= \begin{cases} f[i][j-i](初值，从之前的状态继承，即，f[i][j]==f[i][j-i])\\ min\{f[i-k][j-k]+(s[i]-s[i-k])(i-k)\} \end{cases}\\ 就把i个人分成了两部分，一部分人分到大于1个饼干，一部分分到1个饼干;\\ k表示有k个人分到了一个饼干;\\ (s[i]-s[i-k])表示这k个人的怒气值;\\ (i-k)表示有(i-k)个人分到大于1个饼干. f[i][j]={f[i][j−i](初值，从之前的状态继承，即，f[i][j]==f[i][j−i])min{f[i−k][j−k]+(s[i]−s[i−k])(i−k)}​就把i个人分成了两部分，一部分人分到大于1个饼干，一部分分到1个饼干;k表示有k个人分到了一个饼干;(s[i]−s[i−k])表示这k个人的怒气值;(i−k)表示有(i−k)个人分到大于1个饼干.

• 方案倒推一遍即可

• 刚开始一直疑惑的点：很疑惑过程中的一部分状态转移，例如：$(4,5)\to (5,6)$ 且当k=1时，

  那 $f[5][6]=f[4][5]+g[5]\ast 4(s[5]-s[4]=g[5])$，就给我整不会了，这就十分不河里

#include <bits/stdc++.h>
using namespace std;

typedef pair<int, int> PII; 
const int N=32, M=5005;
PII g[N];
int s[N], ans[N], f[N][M];
signed main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>g[i].first;
		g[i].second=i;
	}
	sort(g+1,g+1+n);
	reverse(g+1,g+1+n);
    // 前缀和计算
	for(int i=1;i<=n;i++) s[i]=s[i-1]+g[i].first;
	memset(f,0x3f,sizeof(f));
	f[0][0]=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=m;j++) // 保证j>=i，每个人至少一个
		{
			f[i][j]=f[i][j-i];
			for(int k=1;k<=i;k++)
			{
				f[i][j]=min(f[i][j],f[i-k][j-k]+(s[i]-s[i-k])*(i-k));
			}
		}
	}
	cout<<f[n][m]<<endl;
	int i=n,j=m,h=0;
	while(i && j)
	{
		if(j>=i && f[i][j]==f[i][j-i]) j-=i, h++;
		else 
		{
			for(int k=1;k<=i && k<=j;k++)
			{
				if(f[i][j]==f[i-k][j-k]+(s[i]-s[i-k])*(i-k))
				{
					for(int u=i;u>i-k;u--) ans[g[u].second]=h+1;
					i-=k; j-=k;
					break;
				}
			}
		}
	}
	for(int i=1;i<=n;i++) cout<<ans[i]<<' ';
    return 0;
}