【CODE[VS]】1501--二叉树最大宽度和高度

题目描述 Description

给出一个二叉树，输出它的最大宽度和高度。

输入描述 Input Description

第一行一个整数n。

下面n行每行有两个数，对于第i行的两个数，代表编号为i的节点所连接的两个左右儿子的编号。如果没有某个儿子为空，则为0。

输出描述 Output Description

输出共一行，输出二叉树的最大宽度和高度，用一个空格隔开。

样例输入 Sample Input

5
2 3
4 5
0 0
0 0
0 0

样例输出 Sample Output

2 3

数据范围及提示 Data Size & Hint

n<16
默认第一个是根节点
以输入的次序为编号
2-N+1行指的是这个节点的左孩子和右孩子
注意：第二题有极端数据！
1
0 0
这题你们别想投机取巧了，给我老老实实搜索！

---

代码

#include <iostream>
using namespace std;

int depth;
int vis[20][2], width[20];

void GetDepth(int data, int var){
    if(var > depth){
        depth = var;
    }
    width[var]++;
    int lchild = vis[data][0];
    int rchild = vis[data][1];
    if(lchild != 0){
        GetDepth(lchild, var + 1);
    }
    if(rchild != 0){
        GetDepth(rchild, var + 1);
    }
}

int main(){
    int i, n, wid = 0;
    cin >> n;
    for(i = 1; i <= n; i++){
        cin >> vis[i][0] >> vis[i][1];
    }
    GetDepth(1, 1);
    for(i = 1; i <= n; i++){
        wid = max(wid, width[i]);
    }
    cout << wid << " " << depth << endl;
    return 0;
}