CodeForces 731C Socks

//并查集的应用
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
int f[N];
int color[N];
vector<int>num[N];
int findx(int x)
{
	return ( x == f[x]? x: f[x] = findx(f[x]) );
}
void merge(int x, int y)
{
	int fx = findx(x);
	int fy = findx(y);
	if (fx != fy)
	f[fy] = fx;
}
int main()
{
	int n, m, k, l, r;
	while (cin >> n >> m >> k)
	{
		for (int i = 1; i <= n; i++)
		{
			f[i] = i;
			cin >> color[i]; // 初始化树根,输入颜色
		}
		for (int i = 0; i < m; i++)
		{
			cin >> l >> r;
			merge(l, r); // 将同一天要穿的袜子合并
		}
		for (int i = 1; i <= n; i++)
		num[findx(i)].push_back(color[i]); // 在每只袜子的数根结点处,压栈它的颜色
		
		int ans = 0;
		for (int i = 1; i <= n; i++)
		{
			int len = num[i].size(), maxx = 0; // len为以i为数根结点的袜子有多少只
			if (len <= 1) //这种情况下,就是自己被合并到别的树根,或者没有和别的袜子组合过,所以树根还是自己,这种情况不需要涂色
			continue;
			map<int, int> cnt;
			cnt.clear();
			for (int j = 0; j < len; j++)
			{
				cnt[num[i][j]]++;
				maxx = max(maxx, cnt[num[i][j]]);
			} //找到颜色最多的袜子有多少个,合并后的并查集里,所有其他颜色的袜子,都需要涂成这种颜色
			ans += len - maxx;
			num[i].clear(); // 用不到了就清空是个好习惯...毕竟第一次用可能是全局变量,初始值全0,但第二次开始,每次都有前面的数据了,这刚好又是多组数据的题目,更是要注意清空,例如cnt就是,用前先 cnt.clear()
		}
		cout << ans << endl;
	}
	
	return 0;
}

原文地址:https://www.cnblogs.com/mofushaohua/p/7789485.html