wikioi1501 二叉樹最大寬度和高度

題目描述 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行指的是這個節點的左孩子和右孩子

#include <stdio.h>
#include <string.h>

int a[1000][2],s[1000];
int i,n,x,y; 

void dfs(int i,int k)
{
	s[k]=s[k]+1;
	if(k>x)	x=k;
	if(a[i][1]!=0)	dfs(a[i][1],k+1);
	if(a[i][2]!=0)	dfs(a[i][2],k+1);
}

int main()
{
	scanf("%d",&n);
	memset(a,0,sizeof(a));
	memset(s,0,sizeof(s));
	for(i=1;i<=n;i++)
		scanf("%d%d",&a[i][1],&a[i][2]);
	x=0;
	dfs(1,1);
	y=0;
	for(i=1;i<1000;i++)
		if(s[i]>y)	y=s[i];
	printf("%d %d",y,x);
	return 0;
}