1540 银河英雄传说

来源：互联网发布：软件测试基础方法编辑：程序博客网时间：2024/06/09 13:42

分析：

类似于带权并查集，又由于更新的时候是从根开始叠加，所以pre[i] 为前面含有的战舰数，每次更新pre[i]

再用一个sum数组保存根节点后面的子节点数目，每次合并时将 pre[尾部舰队根节点的] 设为 sum[被接的舰队的根节点]

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#include <cmath>using namespace std;#define maxn 30000+5int fa[maxn];int sum[maxn];int pre[maxn];int find(int x){    if(x == fa[x])return x;    int f = fa[x];    fa[x] = find(fa[x]);    pre[x] += pre[f];    return fa[x];}void init(){    for(int i = 0 ; i < maxn ; ++i){        fa[i] = i;        sum[i] = 1;    }}int main(){    init();    int T,a,b;    char c[3];    cin >> T;    while(T--)    {        scanf("%s %d %d",&c,&a,&b);        int x = find(a);        int y = find(b);        if(c[0] == 'M'){            fa[x] = y;            pre[x] = sum[y];            sum[y] += sum[x];        }        else        {            if( x == y )                printf("%d\n",abs(pre[b] - pre[a])-1);            else printf("-1\n");        }    }}

0 0