codeforces 731C 并查集+贪心

点击打开链接

利用并查集把颜色要相同的下标放在同一集合中,贪心:找到每个集合中出现最多的颜色 把该集合中剩下的颜色变为该颜色即可 ans+=size-mx

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue> 
#include <vector>
#include <stack>
#include <map>
using namespace std;
const int N=2e5+10;
long long ans=0;
int c[N];
int l,r;
int fa[N];
vector<int> v[N];
int num[N];

int find(int x)
{
	if(fa[x]!=x)
	{
		fa[x]=find(fa[x]);
	}
	return fa[x];
}
void Union(int x,int y)
{
	int fx=find(x),fy=find(y);
	fa[fx]=fy; 
}
int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
		fa[i]=i;
	}
	for(int i=1;i<=m;i++)
	{
		cin>>l>>r; //下标 l,r 
	 	Union(l,r);//把颜色要相同的先放在同一个集合 
	}
	int tot=0;
	for(int i=1;i<=n;i++)
	{
		if(fa[i]==i)
		num[i]=++tot;//集合编号 
	}
	for(int i=1;i<=n;i++)
	{
		v[num[find(i)]].push_back(c[i]);
	}
	for(int i=1;i<=tot;i++)
	{
		int y=v[i].size();
		int mx=0;
		map<int,int> x;
		for(int j=0;j<y;j++)//找到集合中颜色最多的 把剩下的元素变成该颜色即可 
		{
			x[v[i][j]]++;
			mx=max(mx,x[v[i][j]]);
		}
		ans+=(y-mx);
	}
	cout<<ans<<endl;
	
	return 0;
}