并查集 - Socks - CodeForces - 731C

并查集 - Socks - CodeForces - 731C

题意：

$首行输入n,m,k，表示有n条袜子，挑选出m对，共有k种可能的颜色。$

$第二行输入n个数字c_1,c_2,...,c_n，c_i表示第i条袜子的颜色。$

$接着输入m行，每行包括两个数字l和r，表示挑选第l条袜子和第r条袜子。$

$现在可以对袜子进行染色，使得一双袜子的颜色相同。$

$问最少需要对几条袜子染色，可以使得每一组的两条袜子颜色相同。$

$对所有的袜子，只能进行一次染色，染完后就不再更改。$

Examples
Input：

3 2 3
1 2 3
1 2
2 3

Output：

2

Input：

3 2 2
1 1 2
1 2
2 1

Output：

0

数据范围：

$$\begin{gathered} 2\le n\le 200000,0\le m\le 200000,1\le k\le 200000, \\ 1\le c_i\le k, \\ 1\le l_i,r_i\le n,l_i\ne r_i。 \\ Timelimit:2000ms，Memorylimit:262144kB \end{gathered}$$

---

分析：

$首先我们将m对袜子两两之间连一条线，那么同一连通块中的袜子就需要染成相同的颜色。$

$假设数量为S的连通块中，第i种颜色出现的次数最多，出现了s_i次。$

$①、若我们对第i种颜色的所有袜子重新染色，需要进行s_i次操作，$

$同时，剩下的所有袜子颜色也都要改变成对应的颜色。假设改变成第j种颜色。$

$若颜色j不在该连通块中，则总的操作次数就S，即所有袜子颜色都要改变。$

$若颜色j也在该连通块中，那么总的次数就是s_i+(S-s_i-s_j)=S-s_j次。$

$②、若不改变第i种颜色的袜子，将其他颜色的袜子都改变为第i种颜色，则总操作次数为S-s_i。$

$由于第i种颜色出现的次数最多，即s_i>s_j，故第②种操作方式是操作次数最少的。$

$这也很好理解，要使得一堆颜色相同，把除了出现次数最多的颜色C_i以外的其他颜色改变成C_i，这样操作次数最少。$

具体落实：

$①、统计每个连通块中各个颜色出现的次数，就需要先用并查集建立连通关系。$

$②、vector来存储每个集合中的所有颜色，编号即为每个连通块的根节点的编号。$

$③、用map来统计每个连通块中各个颜色出现的次数。$

$④、用整个连通块元素的个数减去出现最频繁的颜色出现的次数，就是该连通块需要操作的最少次数。$

$⑤、对每个连通块求最少操作次数，最后求和即可。$

说明：

$$\begin{gathered} 由于统计每个连通块中个颜色出现的次数时，仅用到少部分下标，故我们用map， \\ 而不是开整型数组，再每次将数组置零，这样会超时。 \end{gathered}$$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<vector>

using namespace std;

const int N=2e5+10;

int n,m,k,idx;
int c[N];
int p[N];
vector<int> V[N];

int Find(int x)
{
    if(p[x]!=x) return p[x]=Find(p[x]);
    return p[x];
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) 
    {
        scanf("%d",&c[i]);
        p[i]=i;
    }
    
    int a,b,pa,pb;
    while(m--)
    {
        scanf("%d%d",&a,&b);
        pa=Find(a),pb=Find(b);
        p[pa]=pb;
    }
    
    for(int i=1;i<=n;i++) V[Find(i)].push_back(c[i]);
    
    int res=0;
    for(int i=1;i<=n;i++)
    {
        int t=0;
        if(V[i].size())
        {
            map<int,int> cnt;
            for(int j=0;j<V[i].size();j++)
            {
                cnt[V[i][j]]++;
                t=max(t,cnt[V[i][j]]);
            }
        }
        res+=(V[i].size()-t);
    }
    
    cout<<res<<endl;
    
    return 0;
}

---