AC日记——二叉树最大宽度和高度 1501 codevs

1501 二叉树最大宽度和高度

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 白银 Silver

题目描述 Description

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

输入描述 Input Description

第一行一个整数n。

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

输出描述 Output Description

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

样例输入 Sample Input

2 3

4 5

0 0

样例输出 Sample Output

2 3

数据范围及提示 Data Size & Hint

n<16

默认第一个是根节点

以输入的次序为编号

2-N+1行指的是这个节点的左孩子和右孩子

注意：第二题有极端数据！

0 0

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

思路：

　　水到不能再水的搜索；

来，上代码：

#include<cstdio>
#include<iostream>

using namespace std;

struct node {
    int from,to,next;
};
struct node edge[1000];

int n,num,head[60],ans_h,w[100],ans_w;

inline void edge_add(int from,int to)
{
    num++;
    edge[num].to=to;
    edge[num].from=from;
    edge[num].next=head[from];
    head[from]=num;
}

void search(int now,int now_)
{
    ans_h=max(now_,ans_h),w[now_]++;
    for(int i=head[now];i!=0;i=edge[i].next) search(edge[i].to,now_+1);
}

int main()
{
    scanf("%d",&n);
    int ch_l,ch_r;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&ch_l,&ch_r);
        if(ch_l) edge_add(i,ch_l);
        if(ch_r) edge_add(i,ch_r);
    }
    search(1,1);
    for(int i=1;i<=ans_h;i++) ans_w=max(ans_w,w[i]);
    printf("%d %d
",ans_w,ans_h);
    return 0;
}