题目描述
有一个 n(n≤106) 个结点的二叉树。给出每个结点的两个子结点编号(均不超过 n),建立一棵二叉树(根节点的编号为 1),如果是叶子结点,则输入 0 0
。
建好这棵二叉树之后,请求出它的深度。二叉树的深度是指从根节点到叶子结点时,最多经过了几层。
输入格式
第一行一个整数 n,表示结点数。
之后 n 行,第 i 行两个整数 l、r,分别表示结点 i 的左右子结点编号。若l=0 则表示无左子结点,r=0 同理。
输出格式
一个整数,表示最大结点深度。
输入输出样例
输入 #1
7 2 7 3 6 4 5 0 0 0 0 0 0 0 0
输出 #1
4
简单求深度题 提供两种基本方法
第一种:链式前向星
第二种: 常规数组 ;l[N]存节点的左孩子 r[N]存节点的右孩子。
第一种方法:
#include<iostream>
#include<cstring>
using namespace std;
const int N = 10000010;
int idx, h[N], e[N], ne[N];
int n;
int max_depth;
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u, int depth) {
if (depth > max_depth) max_depth = depth;
if (h[u] == -1) return;
for (int i = h[u]; ~i; i = ne[i]) {
dfs(e[i], depth + 1);
}
}
int main() {
cin >> n;
memset(h, -1, sizeof h);
for(int i=1;i<=n;i++) {
int l, r;
cin >> l >> r;
if (l != 0)
add(i, l);
if (r != 0) add(i, r);
}
dfs(1,1);
cout << max_depth;
return 0;
}
第二种方法:
#include<iostream>
#include<cstring>
using namespace std;
const int N = 10000010;
int l[N], r[N];
int n;
int max_depth;
void dfs(int u, int depth) {
if (depth > max_depth) max_depth = depth;
if (l[u]) dfs(l[u], depth + 1);
if (r[u]) dfs(r[u], depth + 1);
}
int main() {
cin >> n;
for(int i=1;i<=n;i++) {
int a, b;
cin >> a >> b;
if (a!= 0)
l[i] = a;
if (b != 0) r[i] = b;
}
dfs(1,1);
cout << max_depth;
return 0;
}