Submission #47726296

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define sor(vec) sort(vec.begin(),vec.end())
#define rev(vec) reverse(vec.begin(),vec.end())
#define pb push_back
const int mod = 998244353;

#ifdef LOCAL
#include "debug.cpp"
#else
#define deb(...)
#endif

signed main(){
    ios::sync_with_stdio(false); cin.tie(NULL);
    #ifndef ONLINE_JUDGE
    freopen("error.txt", "w", stderr);
    #endif
    int TC=1;
    while(TC--){
        int n;cin>>n;
        int q;cin>>q;
        unordered_map<int,unordered_set<int>> mp;
        for(int i=1;i<=n;i++){
            int x;cin>>x;
            mp[i].insert(x);
        }
        
        while(q--){
        	
        	int a,b;cin>>a>>b;
        	if(mp[a].size()>mp[b].size()){
        		for(auto it:mp[b]) mp[a].insert(it);
        		
        		swap(mp[b],mp[a]);
        	    mp[a].clear();
                cout<<mp[b].size()<<endl;
        	}else{
        		for(auto it:mp[a]) mp[b].insert(it);
        		
        		mp[a].clear();
                cout<<mp[b].size()<<endl;
        	}
        	
        }
    }
}

Submission Info

Judge Result