链接:https://vjudge.net/problem/Aizu-ALDS1_7_B
思路
《挑战程序设计竞赛2:算法和数据结构》一书第154页的例题。
输出信息从左向右依次为:
- u的结点编号
- u的父皇
- u的hxd
- u的子结点数
- u的深度,
- u的高度
- u的种类
首先这道题使用左子右子存储法(个人称呼)存储结点信息:
struct node
{
int parent,left,right,deep,height,son; //deep:深度 height:高度 son:儿子数
};
对于输出信息,u的深度我在Aizu - ALDS1_7_A:Rooted Trees(有根树的表达)文中提过。只不过那个是左子右兄弟,这里是左子右子,需要改动一下:
(一开始没注意到,W了4次之后才发现这里,不够细啊┭┮﹏┭┮)
//左右子树的深度都要+1
int getdeep(int o,int deep)
{
T[o].deep=deep;
if(T[o].right!=-1)
getdeep(T[o].right,deep+1);
if(T[o].left!=-1)
getdeep(T[o].left,deep+1);
}
这里说一下高度如何求:
int getheight(int o)
{
int h1=0,h2=0;
if(T[o].left!=-1)
h1=getheight(T[o].left)+1;
if(T[o].right!=-1)
h2=getheight(T[o].right)+1;
return T[o].height=max(h1,h2);
}
看图说事:
按照题目中所给要求,上图的5点高度:
0的高度为2;
1的高度为1;
2,3,4高度为0。
可以看出:u点的高度为u在左子树上的高度与右子树上的高度的最大值。
递归就可以解决:
int getheight(int o)
{
int h1=0,h2=0;
if(T[o].left!=-1)
h1=getheight(T[o].left)+1;
if(T[o].right!=-1)
h2=getheight(T[o].right)+1;
return T[o].height=max(h1,h2);
}
废话不多说,上代码!
伊丽莎白!
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+50;
struct node
{
int parent,left,right,deep,height,son;
} T[maxn];
int hxd[maxn];
int getdeep(int o,int deep)
{
T[o].deep=deep;
if(T[o].right!=-1)
getdeep(T[o].right,deep+1);
if(T[o].left!=-1)
getdeep(T[o].left,deep+1);
}
int getheight(int o)
{
int h1=0,h2=0;
if(T[o].left!=-1)
h1=getheight(T[o].left)+1;
if(T[o].right!=-1)
h2=getheight(T[o].right)+1;
return T[o].height=max(h1,h2);
}
int main()
{
int n,id,past,x,y,root;
scanf("%d",&n);
for(int i=0; i<n; i++)
T[i].parent=T[i].left=T[i].right=-1,T[i].son=0;
memset(hxd,-1,sizeof(hxd));
for(int i=0; i<n; i++)
{
scanf("%d",&id);
scanf("%d%d",&x,&y);
if(x!=-1)
{
T[id].left=x;
hxd[x]=y;
T[x].parent=id;
T[id].son++;
}
if(y!=-1)
{
T[id].right=y;
hxd[y]=x;
T[y].parent=id;
T[id].son++;
}
}
for(int i=0; i<n; i++)
if(T[i].parent==-1)
{
root=i;
break;
}
getdeep(root,0);
getheight(root);
for(int i=0; i<n; i++)
{
printf("node %d: parent = %d, sibling = %d, degree = %d, depth = %d, height = %d, ",i,T[i].parent,hxd[i],T[i].son,T[i].deep,T[i].height);
if(T[i].parent==-1)
printf("root\n");
else if(T[i].left==-1&&T[i].right==-1)
printf("leaf\n");
else
printf("internal node\n");
}
}
认真系列×2!