洛谷P15801 [GESP202603 六级] 完全二叉树题解

通过 dfs遍历 计算每个子树的高度（h[u]），并通过dp[u]标记子树是否为完全二叉树，最终统计dp[u]>0的节点数作为答案。

#include<bits/stdc++.h>
using namespace std;
#define N 100086
struct node {
	int l, r;
};
node a[N];
int dp[N], h[N];
int ans, n;
void dfs(int u, int de) {
	h[u] = de;
	if (a[u].l > 0) {
		dfs(a[u].l, de + 1);
		h[u] = max(h[u], h[a[u].l]);
	}
	if (a[u].r > 0) {
		dfs(a[u].r, de + 1);
		h[u] = max(h[u], h[a[u].r]);
	}
// 情况1：a[u].l == 0 && a[u].r == 0)叶子节点左右都为空dp[u]=2视完全二叉树
	if (a[u].l == 0 && a[u].r == 0) dp[u] = 2;
// 情况2：只有左子树，且左子树高度=当前深度+1 dp[u]=1
	else if (a[u].r == 0 && h[a[u].l] == de + 1)dp[u] = 1;
// 情况3：左右子树高度相等且左右都是满完全时赋值dp[u]
	else if (h[a[u].l] == h[a[u].r]) {
		if (dp[a[u].l] == 2 && dp[a[u].r] == 2)dp[u] = 2;
		else if (dp[a[u].l] == 2 && dp[a[u].r] == 1)dp[u] = 1;
	}
// 情况4：左子树比右子树高1仅当右是满、左是完全时赋值dp[u]
	else if (h[a[u].l] == h[a[u].r] + 1) {
		if ((dp[a[u].l] == 1 || dp[a[u].l] == 2) && dp[a[u].r] == 2)dp[u] = 1;
	}
	ans += (dp[u] > 0);
}
int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i].l >> a[i].r;
	}
	dfs(1, -1);
	cout << ans;
	return 0;
}