[xsy2300]好题

Posted 2020-11-08 jefflyy

题意：有一棵树，每个节点有颜色，要找出最小的连通块使得其中的点至少有$k$种不同的颜色，只需输出这个最小连通块的大小

因为$k$很小，所以如果颜色只有$k$种，我们可以直接状压DP，设$f_{i,j}$表示在$i$的子树中包含颜色集合为$j$的最小连通块大小，那么可以枚举$s$的子集$t$，转移即为$f_{x,s}\gets f_{x,t}+f_{u,s-t}\left(u\in son_x\right)$

但颜色最多有$n$种，这时候我们就可以trick一下：随机地把$n$种颜色映射到$k$种颜色再DP，随机多几次错误的概率就很低了

......但我第一次交还是WA了啊==同样的代码再交一次就A了（论一个非洲人的自我修养）

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<time.h>
int min(int a,int b){return a<b?a:b;}
int h[10010],nex[20010],to[20010],c[10010],p[10010],f[10010][32],M,k;
void add(int a,int b){
	M++;
	to[M]=b;
	nex[M]=h[a];
	h[a]=M;
}
void dfs(int fa,int x){
	int i,s,t;
	f[x][p[c[x]]]=1;
	for(i=h[x];i;i=nex[i]){
		if(to[i]!=fa){
			dfs(x,to[i]);
			for(s=0;s<1<<k;s++){
				for(t=s;t;t=(t-1)&s)f[x][s]=min(f[x][s],f[x][t]+f[to[i]][s^t]);
			}
		}
	}
	for(s=0;s<1<<k;s++){
		for(t=s;t;t=(t-1)&s)f[x][t]=min(f[x][t],f[x][s]);
	}
}
int main(){
	srand(time(0));
	int T,n,i,x,y,ans;
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++)scanf("%d",c+i);
	for(i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	T=50;
	ans=n;
	while(T--){
		for(i=1;i<=n;i++)p[i]=1<<(rand()%k);
		memset(f,63,sizeof(f));
		dfs(0,1);
		for(i=1;i<=n;i++)ans=min(ans,f[i][(1<<k)-1]);
	}
	printf("%d",ans);
}