题目:二叉树最大宽度和高度
时间限制: 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;
}