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]);
}