CODE[VS] NO.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

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<queue>
#include<stack>
#include<cstring>
#include<set>
#define INF 0x3f3f3f3f
#define MAX 1000
#define PI 3.1415926

using namespace std;

int main(){
    int n;
    cin >> n;
    int arr[3][25];
    //存储高度，hm和wm分别代表最大深度和宽度
    int h[MAX] = {0}, w[MAX] = {0}, hm = 0, wm = 0;
    //一定从1开始，记住！！！
    h[1] = 1;
    //arr[1][i] 和 arr[2][i]分别代表编号为i的两个子节点
    for(int i = 1; i <= n; i++){
        cin >> arr[1][i] >> arr[2][i];
    }
    //对应的子节点深度加1；若不懂可以将数组表示二叉树的编号写下来
    for(int i = 1; i <= n; i++){
        if(arr[1][i] != 0)
            h[arr[1][i]] = h[i] + 1;
        if(arr[2][i] != 0)
            h[arr[2][i]] = h[i] + 1;
    }
    //每层深度出现一个节点，对应宽度都应加1
    for(int i = 1; i <= n; i++){
        hm = max(hm, h[i]);
        w[h[i]] += 1;
    }
    for(int i = 1; i <= n; i++){
        wm = max(wm, w[i]);
    }
    cout << wm << ' ' << hm << endl;
    return 0;
}