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

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

看看这个：https://www.cnblogs.com/shenben/p/5516842.html

然后理解一下宽度的意思，深度比较好理解了。

二叉树的宽度是指具有节点数最多的那一层的结点个数。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring> 
using namespace std;

int a[30][3],b[30];
int n,x,y,k,wide,deep;

int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d%d",&x,&y);
        a[i][0]=x;
        a[i][1]=y;
        a[x][2]=i;
        a[y][2]=i;
    }
    for(int i=1; i<=n; i++)
    {
        k=1;
        x=a[i][2];
        while(x!=0)
        {
            k++;
            x=a[x][2];
        }
        b[k]++;
        if(k>deep) deep=k;
        if(b[k]>wide) wide=b[k];
    }
    printf("%d %d
",wide,deep);
    return 0;
}

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;

int n,ans1,ans2;
int fa[1002],deep[1002],wide[1002];

struct node
{
    int l,r;
} a[1002];

void dfs(int now)
{
    deep[now]=deep[fa[now]]+1;
    ans2=max(ans2,deep[now]);
    wide[deep[now]]++;
    if(a[now].l ) 
        dfs(a[now].l );
    if(a[now].r )
        dfs(a[now].r );
}

int main()
{
    scanf("%d",&n);
    fa[1]=0;
    for(int i=1; i<=n; ++i)
    {
        scanf("%d%d",&a[i].l ,&a[i].r );
        fa[a[i].l ]=i;
        fa[a[i].r ]=i;
    }
    dfs(1);
    for(int i=1; i<=n; ++i)
        ans1=max(ans1,wide[i]);
    printf("%d %d",ans1,ans2);
}