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

题解

#include <bits/stdc++.h>
using namespace std;
struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x):val(x),left(nullptr),right(nullptr){}
};
int depth(TreeNode* root) {
    if(root==nullptr) return 0;
    return max(depth(root->left),depth(root->right))+1;
}
int main() {
    int n;
    cin>>n;
    unordered_set<int> myset={0};
    vector<TreeNode*> tree={n+1,nullptr};
    for (int i=0;i<n;i++) {
        int a,b;
        cin>>a>>b;
        if(tree[i+1]==nullptr) {
            tree[i+1]=new TreeNode(i+1);
        }
        if(tree[a]==nullptr&&a!=0&&!myset.count(a)) {
            tree[a]=new TreeNode(a);
        }
        if(tree[b]==nullptr&&b!=0&&!myset.count(b)) {
            tree[b]=new TreeNode(b);
        }
        if(a!=0) tree[i+1]->left=tree[a];
        if(b!=0) tree[i+1]->right=tree[b];
        myset.insert(a);
        myset.insert(b);
    }
    cout<<depth(tree[1]);
}