DTOJ#3230. 好题 (hao)

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

有一棵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
*/

猜你喜欢

转载自blog.csdn.net/CSDNzhanghongyu/article/details/109525388