P4913 【深基16.例3】二叉树深度

题目描述

有一个 $n(n\le 10^6)$ 个结点的二叉树。给出每个结点的两个子结点编号（均不超过 $n$），建立一棵二叉树（根节点的编号为 $1$），如果是叶子结点，则输入 0 0。

建好这棵二叉树之后，请求出它的深度。二叉树的深度是指从根节点到叶子结点时，最多经过了几层。

输入格式

第一行一个整数 $n$，表示结点数。

之后 $n$ 行，第 $i$ 行两个整数 $l$、$r$，分别表示结点 $i$ 的左右子结点编号。若 $l=0$ 则表示无左子结点，$r=0$ 同理。

输出格式

一个整数，表示最大结点深度。

样例 #1

样例输入 #1

7
2 7
3 6
4 5
0 0
0 0
0 0
0 0

样例输出 #1

4

思路

代码随想录 104.二叉树的最大深度

代码

DFS版本

这实际上是求根节点的高度，在上面思路的连接中也有递归求深度的代码版本

#include<iostream>
#include<algorithm>
#include<cstring>
#define endl '\n'
using namespace std;

const int N = 1000010;

int n;
int lson[N], rson[N];

int dfs(int t) {
	if (t == 0) {
		return 0;
	}
	// 树的最大深度就是根节点的高度
	// 根节点的高度 = max（左子树的高度，右子树的高度）+ 1
	return max(dfs(lson[t]), dfs(rson[t])) + 1;
}

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n;
	for (int i = 1; i <= n; i++) {
		int l, r;
		cin >> l >> r;
		lson[i] = l;
		rson[i] = r;
	}
	cout << dfs(1) << endl;
	return 0;
}

BFS版本

这求的确实是深度

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define endl '\n'
using namespace std;

const int N = 1000010;

int n;
int lson[N], rson[N];

int bfs(int t) {
	if (t == 0) {
		return 0;
	}
	int depth = 0;

	queue<int> q;
	q.push(t);

	while (q.size()) {
		int cnt = q.size();// 该层的节点数

		depth++;// carl哥的宽搜是一次性就把该层的所有儿子节点全部入队，所以每次while循环，depth++

		for (int i = 0; i < cnt; i++) {
			int x = q.front();
			q.pop();
			if (lson[x]) {
				q.push(lson[x]);
			}
			if (rson[x]) {
				q.push(rson[x]);
			}
		}
	}

	return depth;
}

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n;
	for (int i = 1; i <= n; i++) {
		int l, r;
		cin >> l >> r;
		lson[i] = l;
		rson[i] = r;
	}
	cout << bfs(1) << endl;
	return 0;
}