2676 众里寻花(并查集)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/weixin_43540515/article/details/102572159

有n个彩灯每个彩灯有一种颜色coli(彩灯从1到n标号),有m根细线将它们连接在一起。你想用这些彩灯作出一朵美丽的花,进一步说你想从这m根细线中保留n−1根,且需要保证所有彩灯是连通在一起的。你希望这朵花五彩斑斓,因此你想要连接同颜色彩灯的细线越少越好。请问这个最小值是多少?
数据范围:2≤n≤100,1≤m≤200,1≤coli≤n

输入
第一行两个整数n, m表示彩灯数与细线数。
接下来一行n个数,表示n个彩灯的颜色。
接下来m行,每行两个数u, v,表示u, v这两个彩灯之间有一条细线连接。
输出
输出仅一行,一个整数表示最少多少根细线连接同色灯
输入样例

4 4
1 1 2 2
1 2
2 3
3 4
4 1
输出样例
1
一说到连通就在想dfs,最短路的我简直弱爆了,幸好很快就想到了并查集
题目保证肯定是能把所有花灯连一起的,而又求的是最小连接相同的细线数,那就先把连接不同的细线都用上,看最小能分成多少堆,答案就是堆数-1,因为题目保证肯定能所有连上,就说明不同堆直接肯定有一条线可连

#include<bits/stdc++.h>
using namespace std;
int vis[110],color[110],fa[110];
int find(int x)
{
 if(fa[x]==x) return x;
 return find(fa[x]);	
}
int main()
{	
 ios::sync_with_stdio(false);
 int n,m,u,v,ans=0;
 cin>>n>>m;
 for(int i=1;i<=n;i++)
 {
  cin>>color[i];
  fa[i]=i;	
 }
 for(int i=1;i<=m;i++)
 {
 	cin>>u>>v;
 	if(color[u]!=color[v])
 	 fa[find(u)]=find(v);
 }
 vis[find(1)]=1;
 for(int i=2;i<=n;i++)
  if(vis[find(i)]==0)
  {
  	vis[find(i)]=1;
  	ans++;
  }
 cout<<ans;
 return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43540515/article/details/102572159