【洛谷】P4913二叉树深度

#include <iostream>
#include <cstdio>
using namespace std;

//简单树的遍历题
//需要注意的是，所给数据不一定会按构造顺序给出，故在读入中记录最大深度的解法
//可能会导致所需数据未读入而WA
#define qmax(x, y) (x) > (y)? (x) : (y)
const int mode = 1e6 + 5;
inline void read(int &x)
{
    register int f = 1;
    x = 0;
    char c = getchar();
    while(c > '9' || c < '0')
    {
        if(c == '-') f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9')
    {
        x = (x << 3) + (x << 1) + (c ^ 48);
        c = getchar();
    }
    x *= f;
}

int ans, tr[mode][2];
void dfs(int x, int lev)
{
    ans = qmax(ans, lev);
    if(tr[x][0])
    {
        dfs(tr[x][0], lev + 1);
    }
    if(tr[x][1])
    {
        dfs(tr[x][1], lev + 1);
    }
}

int main()
{
    int n; read(n);
    for(int a, b, i = 1; i <= n; i++)
    {
        read(tr[i][0]); read(tr[i][1]);
    }
    dfs(1, 1);
//    边读边计算是不行的，因为节点可能不会按构造顺序出现
//        if(a) depth[a] = depth[i] + 1;
//        if(b) depth[b] = depth[i] + 1;
//        if(a || b) ans = max(depth[i] + 1, ans);
    printf("%d\n", ans);
    return 0;
}