【算法学习笔记】11.数据结构基础 二叉树初步练习4

时间:2022-04-02 10:24:31

记录二叉树的最大宽度W和深度D

根据输入树的方式,考虑用循环更新的方法来记录,这就需要每一个节点都要有一个所在层的属性,每一层有一个所拥有的宽度的属性

不要去考虑如何记录整个树,没必要

int f[17]={0},w[17]={0};//f中存储着对应节点号的所在层数,方便记录和递增
//w中存储着每层的节点数 

int main()
{
	freopen("in31.txt","r",stdin);
	freopen("out31.txt","w",stdout);
	int n;
	scanf("%d",&n);
	//在输入的过程中就解决问题   
	int wmax=0,fmax=1;
	w[1]=1;f[1]=1;
	for(int i=1;i<=n;i++)
	{
		int l,r;
		cin>>l>>r;//读入第i个节点的左右儿子 
		if(l)//如果有左儿子的话 
		{
			f[l]=f[i]+1;//等于父辈层数加1
			w[f[l]]++;
		}  
		if(r)//如果有右儿子的话 
		{
			f[r]=f[i]+1;//等于父辈层数加1
			w[f[r]]++;
		}		
		fmax=f[i]>fmax?f[i]:fmax;
		wmax=w[f[i]]>wmax?w[f[i]]:wmax; 
	}
	
	cout<<wmax<<" "<<fmax<<endl;
	
}