ZOJ3261_Connections in Galaxy War_并查集::化删为添

题意

n 个星球，m 条道路， q 个命令。战争爆发了， 命令被分为两种：
1.战争破坏了星球 a 和 b 之间的道路。
2. a 星球发出求救信号，输出求救对象。
选择求救对象的规则如下：
1.选择 a 星球能够到达的战斗力（题中给出每个星球的战斗力）最高的星球。
2.如果最高战斗力的星球有两个，取编号较小的那个。
3.求救对象的战斗力必须大于 a 星球。
若符合规则的求救对象存在，输出该编号。否则，输出 -1 。

思路

显然是连通性问题，考虑用并查集来做。
但是并查集能高效地支持 添加边 地操作，而本题主要是删除边地操作。
为此，要把删除边转换成添加边。
具体做法是，把所有的命令保存下来，从后往前倒着执行。即先把被破坏掉的道路全部破坏掉，然后倒着读命令。读到命令 2 ，并查集查找即可。读到命令 1，则向并查集中加入这条边。
如此就把删除边转变成了添加边。

这样做需要记录命令和边。记录边时，由于点数较大，不能采用邻接矩阵。AC代码中是用一个 edge 数组保存边。但这样做的问题是 删除全部被破坏的边时，对应边不容易查找。这里的解决方法是，读入边的时候，用 map 保存边对应的编号。

最后一点，这题的格式比较严格。

题目链接

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3261

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<map>

using namespace std;

typedef pair<int, int> P;

const int maxn = 1e4 + 10;
const int maxm = 2e4 + 10;
const int maxq = 5e4 + 10;

int n, m, q;
char s[27];
int x, y;

int power[maxn];

P edge[maxm];
bool des[maxm];
map<P, int> M;

int qx[maxq], qy[maxq];

int root[maxn];

int ans[maxq], it;

int Uf(int x)
{
    if(root[x] == -1) return x;
    return root[x] = Uf(root[x]);
}

void Union(int x, int y)
{
    x = Uf(x), y = Uf(y);
    if(x == y) return;

    if(power[x] < power[y]) root[x] = y;
    else if(power[x] == power[y] && y < x) root[x] = y;
    else root[y] = x;
}

int main()
{
    //freopen("in.txt", "r", stdin);

    bool first = true;

    while(scanf("%d", &n) != EOF)
    {
        if(first) first = false;
        else printf("\n");

        memset(des, false, sizeof des);
        M.clear();

        for(int i= 0; i< n; i++)
            scanf("%d", power + i);

        scanf("%d", &m);
        for(int i= 0; i< m; i++)
        {
            scanf("%d %d", &x, &y);
            if(x > y) swap(x, y);
            edge[i] = P(x, y);
            M[edge[i]] = i;
        }

        scanf("%d", &q);
        for(int i= 0; i< q; i++)
        {
            scanf(" %s", s);
            if(s[0] == 'q')
            {
                scanf("%d", &qx[i]);
                qy[i] = -1;
            }
            else{
                scanf("%d %d", &x, &y);
                if(x > y) swap(x, y);
                qx[i] = x, qy[i] = y;
                int d = M[P(x, y)];
                des[d] = true;
            }
        }

        memset(root, -1, sizeof root);

        for(int i= 0; i< m; i++)
            if(!des[i]) Union(edge[i].first, edge[i].second);

        it = 0;
        for(int i= q-1; i>= 0; i--)
        {
            if(qy[i] == -1)
            {
                int d = Uf(qx[i]);
                ans[it ++] = power[d] > power[qx[i]] ? d : -1;
            }

            else Union(qx[i], qy[i]);
        }

        for(int i= it - 1; i>= 0; -- i)
            printf("%d\n", ans[i]);
    }

    return 0;
}