小C的周末

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值