标签：fa int 周末 xx mp include define

思路：并查集+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();
    }
}

标签：fa,int,周末,xx,mp,include,define
来源： https://blog.csdn.net/m0_53688600/article/details/117356204

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。