前言
坚持每天做一道算法题,每天学一点数据结构与算法,写思路和题解来加深自己的印象。(好几天没发了,今天补回来)
题目描述
给出每个节点的两个儿子节点,建立一棵二叉树(根节点为 1 ),如果是叶子节点,则输入0 。建好树后希望知道这棵二叉树的深度。二叉树的深度是指从根节点到叶子结点时,最多经过了几层。
最多有 10^6 个结点。
输入格式
第一行一个整数 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
解题思路
如题目的标题所示,就是二叉树的深度,可以通过深度优先搜索,遍历所有结点来实现。
AC代码
#include <iostream>
#include <algorithm> // 为了用max函数,也可以自己写一个
using namespace std;
const int maxn = 1e6; 根据题意的最大值来设置,保险点可以多几个数,例如:1e6+5
int n, res;
struct node {
int left, right; // 结点的左右子树
};
node tree[maxn]; // 建立二叉树
void dfs(int id, int deep) {
if(id == 0) return; // 如果id=0,则就是叶子,换句话说就是到尽头了。
res = max(res, deep); // 取最大的深度
dfs(tree[id].left, deep + 1); // 先找左子树
dfs(tree[id].right, deep + 1); // 再找右子树
}
int main () {
cin >> n; // 初始化结点数
for(int i = 1; i <= n; i++) {
cin >> tree[i].left >> tree[i].right; // 键入数值对二叉树树进行初始化
}
dfs(1, 1); // 从1号结点开始进行深度优先搜索
printf("%d", res); // 输出结果
return 0;
}
总结
递归实现的深度优先搜索,刚开始学的话,最好画图理解一下,看着其实不难,但是得转得过来。(我就转了好久…….)算法之路很漫长~~