洛谷P4913 二叉树深度

枫飘长安2024-04-17 16:50

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

文章目录

[【洛谷P4913 深基16.例3】二叉树深度](#【洛谷P4913 深基16.例3】二叉树深度)

题目描述

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

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

正确代码

c++ 复制代码

#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
int N;
const int MAXN = 1e6+5;
struct node {
    int l,r;
};

void dfs(int num, int deep);

node tree[MAXN];

int ans = 0;

int main() {
    cin >> N;
    for (int i=1; i<=N; i++) {
        cin >> tree[i].l >> tree[i].r;
    }
    dfs(1,1);
    cout << ans  << endl;
    return 0;
}

void dfs(int num, int deep) {
    if (num == 0) return ;
    ans = max(ans,deep);
    dfs(tree[num].l,deep+1);
    dfs(tree[num].r,deep+1);
}