二叉树最大宽度和高度(codevs1501)

来源:互联网 发布:java截取字符串的函数 编辑:程序博客网 时间:2024/06/14 21:44

题目:二叉树最大宽度和高度

时间限制: 1 s

空间限制: 128000 KB

题目描述:
给出一个二叉树,输出它的最大宽度和高度。

输入描述 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(深搜)解决,把某个点的左右儿子节点保存到一个数组里,然后开始dfs,函数中要保留当前节点的层数,另外用一个数组保存当前层数的节点数量(就是宽度嘛)

#include< iostream>
#include< cstdio>
using namespace std;
int t[10000][3],h[10000],n,ans=0,wide=0; //h[k]保存第k层的节点数量,t[i][1],t[i][2]保存编号为i的节点的左右儿子点编号。

void dfs(int i,int k) //i为节点编号,k为节点所在层数
{
h[k]++;//节点所在层数的节点数加上一
if(h[k]>wide) wide=h[k];
if(k>ans) ans=k;//判别最大高度
if(t[i][1]!=0)//等于0就是没有
dfs(t[i][1],k+1);
if(t[i][2]!=0)
dfs(t[i][2],k+1);
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
scanf(“%d%d”,&t[i][1],&t[i][2]);
}
dfs(1,1); //开始dfs,从根节点开始,(根节点在第一层)
printf(“%d %d”,wide,ans);
return 0;
}