有一个划分为 N 列的星际战场,各列依次编号为 1,2,…,N。
有 N 艘战舰,也依次编号为 1,2,…,N,其中第 i 号战舰处于第 i 列。
有 T 条指令,每条指令格式为以下两种之一:
M i j
,表示让第 i 号战舰所在列的全部战舰保持原有顺序,接在第 j 号战舰所在列的尾部。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
带权值的并查集:
p表示父节点,d表示当前节点到所在联通块的根节点的距离,s表示当前联通块的大小
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- #include
- using namespace std;
- typedef long long LL;
- const int N = 3e4 + 5;
- int n;
- int p[N], s[N], d[N];
-
- int find(int a) {
- if (p[a]!=a) {
- int root = find(p[a]);
- d[a] += d[p[a]];
- p[a] = root;
- }
- return p[a];
- }
-
- int main() {
- scanf("%d", &n);
- for (int i = 1; i <= 3e4; i++) {
- p[i] = i;
- s[i] = 1;
- }
- char op[2];
- int a, b;
- while (n--) {
- scanf("%s%d%d", op, &a, &b);
- if (op[0] == 'M') {
- int pa = find(a), pb = find(b);
- if (pa != pb) {
- d[pa] = s[pb];
- s[pb] += s[pa];
- p[pa] = pb;
- }
- }
- else {
- int pa = find(a), pb = find(b);
- if (pa !=pb)cout << "-1" << endl;
- else {
- printf("%d\n", max(0, abs(d[a] - d[b]) - 1));
- }
- }
- }
- return 0;
- }