题解：P11389 [COCI 2024/2025 #1] 等级 / Hijerarhija

思路

因为一棵树的本质是一个图，所以我们可以认为入度为 \(0\) 的节点就是这个树的根。

所以我们统计有几个根，以及是已经作为根的。最后去统计有多少个根就行了。

存储父子关系可以用 unordered_map。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<unordered_map>

using namespace std;

const int N = 3 * 1e5 + 5;
int n, m;
int in[N];
bool root[N];
unordered_map<int, int> fa[N];

int main() {
	scanf("%d", &n);
	for (int i = 1; i < n; i++) {
		int u, v;
		scanf("%d%d", &u, &v);
		fa[u][v] = 0; // 为了更清楚的统计，我们只计算 u -> v 这条边，即 fa[v][u] 
		fa[v][u] = 1;
		in[v]++; // 孩子的入度加 1 
	}
	int cnt = 0;
	for (int i = 1; i <= n; i++) { // 统计 
		if (!in[i]) { // 入度为 0 
			root[i] = 1;
			cnt++;
		}
	}
	if (cnt == 1) printf("DA\n");
	else printf("NE\n");
	scanf("%d", &m);
	for (int i = 1; i <= m; i++) {
		int u, v;
		scanf("%d%d", &u, &v);
		if (fa[u][v]) swap(u, v); // 默认 u 是 v 的父亲，否则交换 
		in[u]++, in[v]--;
		if (root[u]) { // 本来是根的现在不是 
			root[u] = 0;
			cnt--;
		} 
		if (!in[v]) { // 新的根 
			root[v] = 1;
			cnt++;
		}
		if (fa[u][v] == 0) fa[u][v] = 1; // 更新父子关系 
		else fa[u][v] = 0;
		if (fa[v][u] == 0) fa[v][u] = 1;
		else fa[v][u] = 0;
		if (cnt == 1) printf("DA\n");
		else printf("NE\n");
	}
	return 0;
}