AcWing238。銀河英雄伝説

AcWing238。銀河英雄伝説

N列に分割された星間戦場があり、各列には1、2、...、Nの番号が付けられています。

N隻の軍艦があり、これらにも1、2、...、Nの番号が付けられており、i番目の軍艦はi番目の列にあります。

T命令があり、各命令フォーマットは次の2つのいずれかです。

1. M ijは、i番目の戦艦の列にあるすべての軍艦を元の順序で保持し、それらをj番目の戦艦の列の尾に接続することを意味します。

2. C ijは、i番目の戦艦とj番目の戦艦が現在同じ列にあるかどうか、同じ列にある場合は、それらの間に何隻の軍艦があるかを尋ねることを意味します。

次に、一連の命令を処理するプログラムを作成する必要があります。

入力フォーマット

最初の行には整数Tが含まれています。これは、合計でT個の命令があることを意味します。

次のT行、1行に1つの命令、命令にはMijまたはCijの2つの形式があります。

その中で、MとCはコマンドのタイプを示す大文字であり、iとjはコマンドに関係する船番号を示す整数です。

出力フォーマット

プログラムは、各入力命令を順番に分析して処理する必要があります。

M ijの形式の場合は、フリートの配置が変更されたことを意味します。プログラムはこれに注意を払う必要がありますが、情報を出力しないでください。

C ijの形式の場合、プログラムは整数のみを含む1行を出力し、同じ列のi番目の戦艦とj番目の戦艦の間に配置された軍艦の数を示します。i番目の場合戦艦とj番目の戦艦は現在同じ列にありません。-1が出力されます。

データ範囲

$N\le 30000、T\le 500000$

コード

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) {
    
     return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) {
    
     return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 30030;
int n, p[N], s[N], d[N];

int Find(int x) {
    
    
	if (p[x] != x) {
    
    
		int root = Find(p[x]);
		d[x] += d[p[x]];
		p[x] = root;
	}
	return p[x];
}

int main() {
    
    
		for (int i = 1;i < N;++i) {
    
    
			p[i] = i;
			s[i] = 1;
		}
		
		int n;scanf("%d", &n);
		while (n--) {
    
    
			char op[2];int a, b;
			scanf("%s%d%d", op, &a, &b);
			int pa = Find(a);
			int pb = Find(b);
			if (op[0] == 'M') {
    
    
				p[pa] = pb;
				d[pa] = s[pb];
				s[pb] += s[pa];
			}

			else {
    
    
				if (pa == pb)printf("%d\n", max(0, abs(d[a] - d[b]) - 1));
				else printf("-1\n");
			}
		}

	return 0;
}