BZOJ5232[Lydsy2017省队十连测] 好题

原题链接：https://www.lydsy.com/JudgeOnline/problem.php?id=5232

好题

Description

每个人心中对于好题的定义不同，对于小火车来说所谓的好题就是很有趣的题。

有一棵n个节点的树，每个节点都有一种颜色，请找出最小的连通块使得其中的点至少有k种不同的颜色。

Input

第一行两个整数n和k，含义如题所示。

第二行n个正整数，第i个整数表示第i个点的颜色。

接下来n-1行每行两个整数表示一条边。

n<=10000,k<=5，颜色<=n

Output

一行一个整数表示最小的含有至少k个颜色的连通块大小，保证一定存在解。

Sample Input

5 3
1 4 4 2 3
1 2
2 3
3 4
2 5

Sample Output

3

题解

把所有颜色随机映射到$1\sim k$里，多做几次$40$分的状压$dp$，不断取$min$，每次答案正确的机率是$\frac{k!}{k^k}$，所以大概$40$次就出来了。

但非酋是不信这种东西的：

代码

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
const int M=1e4+5;
int head[M],nxt[M<<1],to[M<<1],dp[M][70],col[M],cnt,n,p,ans,mx,tot,final=inf;
vector<int>blo[M];
void add(int f,int t){nxt[++cnt]=head[f],head[f]=cnt,to[cnt]=t;}
void dfs(int v,int f)
{
	dp[v][1<<col[v]-1]=1;
	for(int i=head[v];i;i=nxt[i])
	{
		if(to[i]==f)continue;dfs(to[i],v);
		for(int j=0;j<=mx;++j)
		{
			if((j&(1<<col[v]-1))&&dp[v][j]!=inf)for(int k=0;k<=mx;++k)if(dp[to[i]][k]!=inf)dp[v][j|k]=min(dp[v][j|k],dp[v][j]+dp[to[i]][k]);
			if(__builtin_popcount(j)>=p)ans=min(ans,dp[v][j]);
		}
	}
}
void reset()
{
	ans=inf;
	for(int i=1;i<=n;++i)for(int j=0;j<=mx;++j)dp[i][j]=inf;
	for(int i=1,j,now;i<=tot;++i)for(now=rand()%p+1,j=blo[i].size()-1;j>=0;--j)col[blo[i][j]]=now;
}
void in()
{
	scanf("%d%d",&n,&p);
	for(int i=1,a;i<=n;++i)scanf("%d",&a),blo[a].push_back(i),tot=max(tot,a);
	for(int i=1,x,y;i<n;++i)scanf("%d%d",&x,&y),add(x,y),add(y,x);
}
void ac(){mx=(1<<p)-1;for(int i=1;i<=40;++i)reset(),dfs(1,0),final=min(final,ans);printf("%d",final);}
int main(){srand(1000000009);in(),ac();}