【每日一题】P4913 【深基16.例3】二叉树深度

---

前言

坚持每天做一道算法题，每天学一点数据结构与算法，写思路和题解来加深自己的印象。（好几天没发了，今天补回来）

---

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

题目描述

给出每个节点的两个儿子节点，建立一棵二叉树（根节点为 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;
}

---

总结

递归实现的深度优先搜索，刚开始学的话，最好画图理解一下，看着其实不难，但是得转得过来。（我就转了好久…….）算法之路很漫长~~