Connections in Galaxy War ZOJ - 3261 离线操作+逆序并查集 并查集删边

#include<iostream>
#include<cstring>
#include<stdio.h>
#include<map>
#include<vector>
#define cle(a) memset(a,0,sizeof(a))
using namespace std;
const int N=+;
int w[];
bool cmp(int a,int b){
    return a>b;
}
int n,q,m,fa[N],arr[][],ans[];
vector<pair<int,int> >vp;
map<pair<int,int>,int>mp;
void init(){
    for(int i=;i<=n+;i++)fa[i]=i;
    mp.clear();
    vp.clear();
    cle(ans);
    cle(arr);
}
int find(int x){
    if(fa[x]!=x)
        fa[x]=find(fa[x]);
    return fa[x];
}
void Union(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x==y)
        return;
    if(w[x]>w[y])
        fa[y]=x;
    else if(w[x]<w[y])
        fa[x]=y;
    else{    //这里 的判断要注意
        if(x<y)
            fa[y]=x;
        else
            fa[x]=y;
    }
}
int main()
{
    int mark=;
    while(cin>>n)
    {
        if(mark)
            printf("\n");
        init();
        //每个点的价值
        for(int i=;i<n;i++)
            scanf("%d",&w[i]);
        scanf("%d",&m);
        //边
        for(int i=;i<=m;i++)
        {
            scanf("%d%d",&arr[i][],&arr[i][]);
            //大的在前面，为后面建边做准备
            if(arr[i][]>arr[i][])
                swap(arr[i][],arr[i][]);
        }
        scanf("%d",&q);
        char s[];
        int t,p;
        for(int i=;i<=q;i++)
        {
            scanf("%s",s);
            if(s[]=='q')
            {
                scanf("%d",&t);
                vp.push_back({t,-});
            }
            else{
                scanf("%d%d",&t,&p);
                if(t>p)
                    swap(t,p);
                //标记已经建过边了，相当于标记要被拆掉
                mp[{t,p}]=;
                vp.push_back({t,p});
            }
        }
        //先把不会被拆掉的边建上
        for(int i=;i<=m;i++)
            if(!mp[{arr[i][],arr[i][]}])
                Union(arr[i][],arr[i][]);
        //然后从后往前遍历操作
        //如果是查询，直接做
        //如果是拆边，就建上
        int j=;
        for(int i=vp.size()-;i>=;i--){
            int a=vp[i].first;
            int b=vp[i].second;
            //如果是查询
            if(b==-)
            {
                //找到父节点
                int c=find(a);
                //如果相同，就是-1
                if(w[a]>=w[c])
                    c=-;
                //保存答案
                ans[j++]=c;
            }
            else
                Union(a,b);
        }
        for(int i=j-;i>=;i--)
            printf("%d\n",ans[i]);
        mark=;
    }
    return ;
}