Gym 100703J 并查集

点击打开链接

题意：给了n个人，m对朋友关系，和p个询问，然后接下来的第一行是第二个人到最后一个人参加游戏的最大时间，下面p个是每个游戏的持续时间，然后底下的是m对关系，一个人参加游戏的最大时间若小于游戏的持续时间，则这个人离开关系图，也就是他以后不能传递关系了，只有有关系的人才可以参加游戏

思路：对于每个人我们可以先处理出他可以参加到哪个游戏，也就是什么时候离开关系图，然后不会离开的所有人都打上标记，p个询问从后面开始，若有点是在i离开的，则将与i有关系的且没离开的相连，就像是一个空白的关系图，将每个人请回来，然后有关系的连接，这部分就用并查集就可以了

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
int f[maxn],cnt[maxn];
ll ans[maxn];
int find1(int x){if(x!=f[x]) f[x]=find1(f[x]);return f[x];}
void unite(int a,int b){
    int aa=find1(a);
    int bb=find1(b);
    if(aa==bb) return ;
    f[aa]=bb;cnt[bb]+=cnt[aa];
}
vector<int>G[maxn],GG[maxn];
struct edge{
    int val,id;
}num[maxn];
bool cmp(const edge &a,const edge &b){return a.val<b.val;}
int vis[maxn],fvis[maxn],Q[maxn];
int main(){
    int n,m,q,u,v;
    while(scanf("%d%d%d",&n,&m,&q)!=-1){
        for(int i=0;i<=n;i++) G[i].clear(),GG[i].clear(),vis[i]=0,fvis[i]=0,f[i]=i,cnt[i]=1;
        for(int i=2;i<=n;i++) scanf("%d",&num[i].val),num[i].id=i;
        for(int i=0;i<q;i++) scanf("%d",&Q[i]);
        for(int i=0;i<m;i++){
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        sort(num+2,num+n+1,cmp);
        for(int i=0,j=2;i<q;i++){
            while(j<=n&&num[j].val<Q[i]){
                vis[num[j].id]=1;
                GG[i].push_back(num[j].id);j++;
            }
        }
        fvis[1]=1;
        for(int i=2;i<=n;i++){
            if(vis[i]==0){
                for(unsigned int j=0;j<G[i].size();j++){
                    int t=G[i][j];
                    if(fvis[t]) unite(i,t);
                }
                fvis[i]=1;
            }
        }
        for(int i=q-1;i>=0;i--){
            ans[i]=(ll)cnt[find1(1)]*(ll)(Q[i]);
            for(unsigned int j=0;j<GG[i].size();j++){
                int t=GG[i][j];
                for(unsigned int l=0;l<G[t].size();l++){
                    int tmp=G[t][l];
                    if(fvis[tmp]) unite(t,tmp);
                }
                fvis[t]=1;
            }
        }
        for(int i=0;i<q-1;i++) printf("%I64d ",ans[i]);
        printf("%I64d\n",ans[q-1]);
    }
    return 0;
}