小C的周末----并查集+二维map+思维

题目链接

算是比较奇怪的并查集了

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
//#define int long long
const int N = 100010;
int p[N];
map<int,int>mp[N];
int n,m,q;
int vis[N],cnt[N];
int find(int x)
{
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}
void init()
{
    for(int i=1;i<=n;i++)p[i]=i,mp[i].clear();
}
signed main()
{
    while(cin>>n>>m>>q)
    {
        init();
        memset(cnt,-1,sizeof cnt);
        memset(vis,0,sizeof vis);
        for(int i=1;i<=n;i++)
        {
            int x;scanf("%d",&x);
            vis[x]++;
            mp[i][x]++;
        }
        for(int i=1;i<=q;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            int pa=find(a),pb=find(b);
            if(pa==pb)continue;
            if(mp[pa].size()>mp[pb].size())swap(pa,pb);
            p[pa]=pb;
            for(auto it : mp[pa])
            {
                mp[pb][it.first]+=it.second;
                if(mp[pb][it.first]==vis[it.first]&&cnt[it.first]==-1)
                {
                    cnt[it.first]=i;
                }
            }
            
        }
        for(int i=1;i<=m;i++)
        {
            if(vis[i]==1)cout<<0<<endl;
            else cout<<cnt[i]<<endl;
        }   
    }
}