【並查集】【DFS序】Idiot 的間諜網絡

Idiot 的間諜網絡

題目大意

就是有三種類型
類型1 x y：y 成爲x 的直接領導；
類型2 x：x 策劃了一起行動，然後上報其直接領導審批，之後其直接領導再上報其直接領導的直接領導審批，以此類推，直到某個特工審批後不再有直接領導；
類型3 x y：詢問x 是否直接策劃或審批過第y 次特別行動。所有特別行動按發生時間的順序從1 開始依次編號
（有就輸出YES 否則輸出NO）

輸入

一行，n 和 m 表示人數和多少種事件

輸出

每一次詢問後的答案

輸入樣例

6 12
2 1
1 4 1
3 4 1
1 3 4
2 3
3 4 1
2 3
3 4 2
3 1 1
3 1 3
3 1 2
1 2 4

輸出樣例

NO
NO
YES
YES
YES
YES

解題思路

這道題用並查集和 $DFS$ 序來做，對於一個點是否爲另一個點祖先的詢問，其實可以直接使用 dfs 序來解決
操作1就並查集確立祖先，這樣子操作2的時候就可以看他們的祖先是否相同。
相同就是連通的，否則不連通

程序如下

#include<cstdio>
#include<algorithm>

using namespace std;

struct note {
	int to, next;
}e[500001];
struct read {
	int what, num, who;
}a[500001];
int n, m, le[500001], z[500001], x[500001], y[500001], k, kk, kkk, in[500001], out[500001], father[500001];
bool fa[500001], ans[500001];

bool cmp(read x, read y) {
	return x.what < y.what;//比較大小
}

void dfs(int now) {//DFS序
	in[now] = ++in[0];
	for (int i = le[now]; i; i = e[i].next) {
		dfs(e[i].to);
	}
	out[now] = ++out[0];
}

int find(int now) {//查找是否爲父親
	if (father[now] == now) return now;
	return father[now] = find(father[now]);
}

int main() {
	scanf("%d %d", &n, &m);
	
	for (int i = 1; i <= m; i++) {
		scanf("%d", &z[i]);
		if (z[i] == 1) {
			scanf("%d %d", &x[i], &y[i]);
			e[++k] = (note){x[i], le[y[i]]}; le[y[i]] = k;//建鄰接表
			fa[x[i]] = 1;//找出根節點
		}
		else if (z[i] == 2) {
			scanf("%d", &x[i]);
		}
		else if (z[i] == 3) {
			scanf("%d %d", &x[i], &y[i]);
			a[++kk] = (read){y[i], ++kkk, x[i]};//記錄
		}
	}
	
	sort(a + 1, a + kk + 1, cmp);//排序任務
	
	for (int i = 1; i <= n; i++) {
		father[i] = i;
		if (!fa[i]) {
			dfs(i);
		}
	}
	
	int what_num = 0, what = 1;//初始化
	for (int i = 1; i <= m; i++) {
		if (z[i] == 1) father[find(x[i])] = find(y[i]);//連線
			else if (z[i] == 2) {
				what_num++;
				while (what <= kk && a[what].what <= what_num) {
					if (find(a[what].who) == find(x[i]) && in[a[what].who] <= in[x[i]] && out[a[what].who] >= out[x[i]])//判斷是否連通
						ans[a[what].num] = i;//標記
					what++;
				}
			}
	}
	
	for (int i = 1; i <= kk; i++)
		if (ans[i]) printf("YES\n");
			else printf("NO\n"); 
	
	return 0;
}