某 SCOI 模拟赛 T1 a【DP】

题意

有 $n$ 个单词，每个单词出现 $c_i$ 次，现用 - 与 . 给单词编码，要求任意一个单词的编码不是另一个的前缀。设 - 的权值为 2，. 的权值为 1，最小化所有单词的权值和。$n\le 750$。

题解

假设我们已经建好了所有单词的字典树，显然出现次数越多的单词应该挂在越浅的叶子（下文深浅均指带权的深浅，- 边长度为 2，. 边长度为 1）。故考虑先将 $c_i$ 排序，并 DP。

设 $f[i,j,k,l]$ 表示前 $i$ 层的树已经确定，并挂上了前 $j$ 个单词，这一层有 $k$ 个不确定的、下一层有 $l$ 个不确定的点的最优情况，接着枚举这一层有多少个点要挂上单词，而没挂上单词的节点会贡献到下一层和下下层。这个 DP 是 $O(n^5)$ 的。

考虑优化：首先 $i$ 是没用的，扔了。我们考虑每个 $j$，它要么挂在这一层，由 $f[j,k,l]$ 转移到 $f[j+1,k-1,l]$；要么留给之后，由 $f[j,k,l]$ 转移到 $f[j,l+k,k]$，并且所有没被挂的单词深度会加 1，加上对应的 $c_i$ 之和。这个 DP 是 $O(n^3)$ 的。由于 $j,k$ 的范围到不了满的 $i$，所以有约 $\frac{1}{6}$ 的常数，可以通过。

代码：

#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const int N=760;
int a[N];
long long f[2][N][N];
int s[N];

int main(){
	int n=getint();
	for(int i=1;i<=n;i++){
		a[i]=getint();
	}
	sort(a+1,a+n+1,greater<int>());
	for(int i=n;i>=1;i--)s[i]=s[i+1]+a[i];
	int t=0;
	memset(f,0x7f,sizeof(f));
	f[0][1][1]=s[1];
	for(int i=1;i<=n;i++){
		memset(f[t^1],0x7f,sizeof(f[t^1]));
		for(int j=0;j<=n-i+1;j++){
			for(int k=0;j+k<=n-i+1;k++){
				f[t][j+k][j]=min(f[t][j+k][j],f[t][j][k]+s[i]);
			}
		}
		for(int j=1;j<=n-i+1;j++){
			for(int k=0;j+k<=n-i+1;k++){
				f[t^1][j-1][k]=min(f[t^1][j-1][k],f[t][j][k]);
			}
		}
		t^=1;
	}
	long long ans=0x7f7f7f7f7f7f7f7fll;
	for(int i=0;i<=n;i++)for(int j=0;j<=n;j++)ans=min(ans,f[t][i][j]);
	cout<<ans<<endl;
	return 0;
}