每个人心中对于好题的定义不同,对于小火车来说所谓的好题就是很有趣的题。
有一棵n个节点的树,每个节点都有一种颜色,请找出最小的连通块使得其中的点至少有k种不同的颜色。
题解:随机化。。。
随机30次就差不多了
#include<bits/stdc++.h>
#define N 10003
using namespace std;
inline int read(){
int x=0;char s=getchar();
while(s<'0'||s>'9')s=getchar();
while(s>='0'&&s<='9'){
x=(x<<3)+(x<<1)+s-'0';s=getchar();}
return x;
}
int n,k,ans,limi,tot,head[N],nex[N<<1],ver[N<<1],c[N],f[N][1<<5],siz[N];
inline void add(int x,int y){
nex[++tot]=head[x];head[x]=tot;ver[tot]=y;
}
int g[N][1<<5];
void dfs(int x,int las){
f[x][1<<(c[x]-1)]=1;
//f[x][0]=0;
siz[x]=1;
for(int i=head[x];i;i=nex[i]){
int y=ver[i];
if(y==las)continue;
dfs(y,x);
siz[x]+=siz[y];
for(int j=0;j<(1<<limi);++j)g[x][j]=f[x][j];
for(int j=0;j<(1<<limi);++j){
for(int h=0;h<(1<<limi);++h){
g[x][j|h]=min(g[x][j|h],f[x][j]+f[y][h]);
}
}
for(int j=0;j<(1<<limi);++j)f[x][j]=min(f[x][j],g[x][j]);
}
ans=min(ans,f[x][(1<<limi)-1]);
}
int v[N],fa[N],pre[N],to[N];
inline int get(int x){
if(fa[x]==x)return x;
return fa[x]=get(fa[x]);
}
inline int ran(int x){
return rand()%x+1;
}
int main(){
n=read(),k=read();
int maxn=0;
for(int i=1;i<=n;++i)c[i]=read(),maxn=max(maxn,c[i]),pre[i]=c[i];
for(int i=1;i<n;++i){
int x=read(),y=read();
add(x,y);add(y,x);
}
ans=0x3f3f3f3f;
limi=k;
int t=30;ans=0x3f3f3f3f;
while(t--){
for(int i=1;i<=n;++i)to[i]=ran(k);
for(int i=1;i<=n;++i)c[i]=to[pre[i]];
memset(f,0x3f,sizeof(f));
dfs(1,0);
}
printf("%d\n",ans);
return 0;
}
/*
5 3
1 1 1 2 3
1 2
2 3
3 4
2 5
*/