【AcWing 238】 银河英雄传说 并查集 深度记录

有一个划分为N列的星际战场，各列依次编号为1,2,…,N。

有N艘战舰，也依次编号为1,2,…,N,其中第i号战舰处于第i列。

有T条指令，每条指令格式为以下两种之一：

1、M i j，表示让第i号战舰所在列的全部战舰保持原有顺序，接在第j号战舰所在列的尾部。

2、C i j，表示询问第i号战舰与第j号战舰当前是否处于同一列中，如果在同一列中，它们之间间隔了多少艘战舰。

现在需要你编写一个程序，处理一系列的指令。

输入格式
第一行包含整数T，表示共有T条指令。

接下来T行，每行一个指令，指令有两种形式：M i j或C i j。

其中M和C为大写字母表示指令类型，i和j为整数，表示指令涉及的战舰编号。

输出格式
你的程序应当依次对输入的每一条指令进行分析和处理：

如果是M i j形式，则表示舰队排列发生了变化，你的程序要注意到这一点，但是不要输出任何信息；

如果是C i j形式，你的程序要输出一行，仅包含一个整数，表示在同一列上，第i号战舰与第j号战舰之间布置的战舰数目，如果第i号战舰与第j号战舰当前不在同一列上，则输出-1。

数据范围
N≤30000,T≤500000
输入样例：
4
M 2 3
C 1 2
M 2 4
C 4 2
输出样例：
-1
1

题意：如题

思路：

并查集的题意很明显。只不过多问了一个深度问题。两个人之间的距离可以转化成两个人到达根节点的距离的差值（绝对值）。那么只需要在并查集合并的时候，若是a->b，则a结合到b的时候a的深度即是b的结点数量。结合完之后b的结点数量更新，加上a的结点数量。然后a的结点在更新深度的时候（get()函数中）就也加上a的深度变化即可。 如原本有c->a ，那么 在a->b之后，假设a的深度变为10，则c->a->b的深度就是depth（c->a）+depth（a->b）= 11。这在递归的时候加一句话即可。详见代码

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
const int maxn = 3e4+5;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'||ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };
ll depth[maxn];
ll fa[maxn];
ll num[maxn];
ll n;
void init()
{
    rep(i,1,maxn-3)
    fa[i] = i , depth[i] = 0, num[i] = 1;;
}

ll get(ll x)
{
    if(fa[x]==x) return fa[x] = x;
    ll t = fa[x];
    fa[x] = get(fa[x]);
    depth[x] += depth[t];
    return fa[x];
}

void merge(ll x, ll y)
{
    ll a = get(x);
    ll b = get(y);
    if(a!=b)
    {
        fa[a] = b;
        depth[a] = num[b];
        num[b] += num[a];
    }
}


int main()
{
    FAST;
    init();
    int t;
    cin>>t;
    while(t--)
    {
        string s;
        cin>>s;
        if(s=="M")
        {
            ll x,y;
            cin>>x>>y;
            merge(x,y);
        }
        else
        {
            ll x,y;
            cin>>x>>y;
            ll a = get(x); ll b = get(y);
            if(a==b)
            {
                cout<<abs(depth[x] - depth[y])-1<<endl;
            }
            else  cout<<-1<<endl;
        }
    }
    return 0;
}