【随机化】【树形DP】【状压DP】BZOJ5232[Lydsy2017省队十连测]好题

分析

k很小。。。
如果所有颜色都在k以内，那么直接状压即可。

如果不在呢？那么随机将某种颜色，映射到k以内的另一个颜色。换言之，我们认为所有映射到同一位置的颜色相同。
显然，除了最优策略，其他的情况一定不优于最优策略。

那么，一次的正确性显然就是：$\frac{k!}{k^k}$即最优的k个颜色：能找出最优解的情况/总的映射情况。
一次的正确率大约为0.03
做个50次左右就接近1了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#define SF scanf
#define PF printf
#define MAXN 10010
using namespace std;
int n,k,col[MAXN],cnt[MAXN],sum,cntc[MAXN],ans;
vector<int> a[MAXN];
int dp[MAXN][35],colx[MAXN];
void dfs(int x,int fa){
	dp[x][(1<<(col[x]-1))]=1;
	for(int i=0;i<int(a[x].size());i++){
		int u=a[x][i];
		if(u==fa)
			continue;
		dfs(u,x);
		for(int mask1=(1<<k)-1;mask1>=0;mask1--)
			for(int mask2=0;mask2<(1<<k);mask2++){
				if(dp[x][mask1]>n||dp[u][mask2]>n)
					continue;
				dp[x][mask1|mask2]=min(dp[x][mask1|mask2],dp[x][mask1]+dp[u][mask2]);
			}
	}
	ans=min(ans,dp[x][(1<<k)-1]);
}
int c2t[MAXN];
int main(){
	srand(20010919);
//	freopen("hao.in","r",stdin);
//	freopen("hao.out","w",stdout);
	SF("%d%d",&n,&k);
	int maxv=0;
	for(int i=1;i<=n;i++)
		SF("%d",&colx[i]);
	int u,v;
	for(int i=1;i<n;i++){
		SF("%d%d",&u,&v);
		a[u].push_back(v);
		a[v].push_back(u);
	}
	ans=n;
	int tims=50;
	while(tims--){
		memset(dp,0x3f3f3f3f,sizeof dp);
		for(int i=1;i<=n;i++)
			c2t[i]=rand()%k+1;
		for(int i=1;i<=n;i++)
			col[i]=c2t[colx[i]];
		dfs(1,0);
	}
	PF("%d",ans);
}