CODE[vs] 天梯 1501 二叉树最大宽度和高度

二叉树最大宽度和高度
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 白银 Silver
题解

题目描述 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

这题你们别想投机取巧了，给我老老实实搜索！

思路：没啥思路，我就是造出一棵树然后dfs搜索，求出宽度和高度。ps:说实话，这个宽度我搞了好久，写重复了一个，弄的答案错了！

代码：（略长）

#include<iostream>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;

struct tree{//树
    int num;
    struct tree *leftT;
    struct tree *rightT;
};

struct tree *p;//指向当前需要的树节点

void find(struct tree *tr, int i){
    if (tr->num == i){
        p = tr; return;
    }
    if (tr->leftT)find(tr->leftT, i);
    if (tr->rightT)find(tr->rightT, i);
}

//计算深度 
int deepth(struct tree *tr){
    if (!tr)return 0;
    int m = max(deepth(tr->leftT)+1, deepth(tr->rightT) + 1);
    return m;
}

int a[100], maxs = 1;//默认maxs的宽度为1，为根节点
//计算宽度
void width(struct tree * tr, int i){
    if (!tr)return;
    if (tr->leftT)a[i + 1]++;
    if (tr->rightT)a[i + 1]++;

    if (maxs<a[i + 1])maxs = a[i + 1];
    width(tr->leftT, i + 1);
    width(tr->rightT, i + 1);
}


void dfs(struct tree *tr, int i){
    int n, m;
    cin >> n >> m;
    if (n == 0 && m == 0)return;
    find(tr, i);

    if (n > 0){
        p->leftT = new tree;
        p->leftT->num = n;
        p->leftT->leftT = NULL;
        p->leftT->rightT = NULL;
    }

    if (m > 0){
        p->rightT = new tree;
        p->rightT->num = m;
        p->rightT->leftT = NULL;
        p->rightT->rightT = NULL;
    }

}
int main(){
    a[1] = 1;
    int n;
    cin >> n;

    struct tree *tr = new tree;
    tr->num = 1;
    tr->leftT = NULL;
    tr->rightT = NULL;
    for (int i = 1; i <= n; i++){
        dfs(tr, i);
    }
    int deep = deepth(tr);
    width(tr, 1);
    cout << maxs << " " << deep << endl;

    return 0;
}