问题:
已知一颗树的层次序列及每个节点的度, 编写算法构造这个树的孩子兄弟链表
分析:
由于层次遍历的顺序可以根据节点的度去充分地遍历每一个节点的兄弟,所以可以每次都把每一层节点的兄弟关系先链接好,并且由于父子关系只能查找一次,所以按照层序序列经过节点时,需要把其孩子节点链接上。
所以按照层序次序遍历节点时,链接上孩子节点,并且链接上该节点的全部兄弟节点。
int length = 9;
int levelOrder[length] = {3, 2, 1, 5, 4, 6, 8, 7, 9};
int degree[length] = {3, 2, 0, 1, 1, 0, 1, 0, 0};
例如:
i=0
时,访问levelOrder[0]
==访问3,3的度为3,所以3是有三个孩子的,
第一个孩子2的id是1,所以3指向数组levelOrder[1]
,也就是2,
此时由于2节点有兄弟,且根据3的度知道,2有2个兄弟,绑定2的兄弟,
根据层序遍历的定义,2,1,5肯定是一个层的,而且他们的id是相邻的1,2,3
,所以绑定levelOrder[1]
的右兄弟为levelOrder[2]
,levelOrder[2]
的右兄弟为levelOrder[3]
,
此时已经遍历到了节点i=1, 也就是访问levelOrder[1]
==访问2,
节点2的度为2,于是右孩子为levelOrder[4]
,
因为在绑定兄弟的时候,前面的节点都访问过了,
而levelOrder[4]
的兄弟是levelOrder[5]
,绑定他们之间的关系,
以此类推,
下面的图说明了这个过程:
算法过程图解:
函数调用:
#include "Tree.h"
int main()
{
int length = 9;
int levelOrder[length] = {3, 2, 1, 5, 4, 6, 8, 7, 9};
int degree[length] = {3, 2, 0, 1, 1, 0, 1, 0, 0};
BroTreeHead btr = GetRootArray2(levelOrder, degree, length);
VisitForestHead(btr);
}
代码实现:
/***
* 已知一颗树的层次序列及每个节点的度,
* 编写算法构造这个树的孩子兄弟链表
*
* 思路:类似于孩子表示法转为孩子兄弟链表
* 第一层是根,下一层的首个节点是根的孩子,
* 由于已知道每个节点的度,所以可以得到每个节点的孩子排列和孩子数目
*/
BroTreeHead GetRootArray2(int levelOrder[], int degree[], int length)
{
BroTreeNode *p = new BroTreeNode[length];
for (int i = 0; i < length; i++)
{
p[i].data = levelOrder[i];
p[i].lchild = p[i].rbrother = NULL;
}
int d;
int nodeId = 0;
for (int i = 0; i < length; i++)
{
d = degree[i]; //度为d
if (d) //有孩子
{
nodeId++;
p[i].lchild = &p[nodeId];
for (int j = 2; j <= d; j++) //绑定好全部的兄弟节点
{
nodeId++;
p[nodeId - 1].rbrother = &p[nodeId];
}
} //否则无孩子
}
return &p[0];
}