思路:并查集+map
mp[i][j]表示第i个集合玩第j个游戏的人数,当人数达到要求时,记录答案,因为普通二维数组开不到1e5所以用开map二维数组
注意:
集合合并时小集合合并到大集合里
开longlong会re
达不到要求输出-1
#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define mst(x,y) memset(x,y,sizeof(x))
#define all(v) v.begin() , v.end()
#define pb(v) push_back(v)
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
//#define int long long
#define lson p*2,l,mid
#define rson p*2+1,mid+1,r
typedef long long ll;
const int N = 1e5+10;
int n,m,q;
int fa[N];
int ans[N];
int num[N];;//游戏需要的人数
map<int ,int > mp[N];//第i个集合玩j游戏的人数
int find(int x)
{
return fa[x] ==x ? x:fa[x] = find(fa[x]);
}
void solve()
{
_for(i,0,n) fa[i]=i;
_for(i,1,q)
{
int x,y;
cin>>x>>y;
int xx = find(x);
int yy = find(y);
//小集合合并到大集合
if( mp[xx].size() < mp[yy].size() ) swap(xx,yy);
fa[yy] =xx;
for(auto it:mp[yy])
{
mp[xx][it.first] += it.second;//大集合加上小集合人数
if( mp[xx][it.first] == num[it.first])
ans[it.first] = i;//保存答案
}
}
_for(i,1,m)
{
if( num[i]==1) cout<<0<<endl;//单机游戏
else cout<<ans[i]<<endl;
}
}
signed main()
{
//!!!!!!!!!!!!!!!!!!!!!!
// freopen("data.txt","r",stdin);
//!!!!!!!!!!!!!!!!!!!!!!
IOS;
while( cin>>n>>m>>q )
{
_for(i,1,n) mp[i].clear();
mst(num,0);
mst(ans,-1);
_for(i,1,n)
{
int x;cin>>x;
mp[i][x]++;
num[x]++;
}
solve();
}
}