洛谷P1196

题目描述

在这次决战中，他将巴米利恩星域战场划分成30000列，每列依次编号为1, 2, …,30000。之后，他把自己的战舰也依次编号为1, 2, …, 30000，让第i号战舰处于第i列，形成“一字长蛇阵”，诱敌深入。这是初始阵形。当进犯之敌到达时，杨威利会多次发布合并指令，将大部分战舰集中在某几列上，实施密集攻击。合并指令为M i j，含义为第i号战舰所在的整个战舰队列，作为一个整体接至第j号战舰所在的战舰队列的尾部。显然战舰队列是由处于同一列的一个或多个战舰组成的。合并指令的执行结果会使队列增大。 然而，老谋深算的莱因哈特早已在战略上取得了主动。在交战中，他可以通过庞大的情报网络随时监听杨威利的舰队调动指令。
在杨威利发布指令调动舰队的同时，莱因哈特为了及时了解当前杨威利的战舰分布情况，也会发出一些询问指令：C i j。该指令意思是，询问电脑，杨威利的第i号战舰与第j号战舰当前是否在同一列中，如果在同一列中，那么它们之间布置有多少战舰。

输入样例#1：

4
M 2 3
C 1 2
M 2 4
C 4 2

输出样例#1：

-1
1

分析

对于每个点，分别记录所属链的头结点、该点到头结点的距离以及它所在集合的大小。每次合并将y接在x的尾部，改变y头的权值和所属链的头结点，同时改变x的尾节点。

#include<cstdio>
using namespace std;
int father[50000],b[50000],c[50000],n;
int find(int v)
{
    int a;
    if (father[v]==0)
        return v;
    else
    {
        a=find(father[v]);
        b[v]=b[father[v]]+b[v];
        father[v]=a;
        return father[v];
    }
}
int abs(int p)
{
    if (p<0) return -p;
    else return p;
}
int sss(int x,int y)
{
    int l,r;
    l=find(x);r=find(y);
    father[l]=r;
    b[l]=b[l]+c[r];
    c[r]=c[l]+c[r];
}
int main()
{
    int j,k;
    char s;
    for (int i=1;i<=50000;i++) c[i]=1;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%s%d%d",&s,&j,&k);
        if (s=='M')
            if (find(j)!=find(k))
                sss(j,k);
            else j=1;
        else
            if (find(j)==find(k))
                printf("%d\n",abs(b[k]-b[j])-1);
            else printf("-1\n");
    }
}