链接二叉树中同一层的节点

最新推荐文章于 2021-04-14 00:46:58 发布

转载最新推荐文章于 2021-04-14 00:46:58 发布 · 1.7k 阅读

算法专栏收录该内容

376 篇文章

订阅专栏

本文深入探讨了一种解决树节点层级链接问题的算法，通过递归方式实现，确保每个节点都能与其同一层的邻居节点建立连接。重点介绍了如何避免使用先左再右的递归方式，转而采用从右到左的递归策略来解决查找同层节点孩子的问题，以及在链接节点时的技巧。

这个题目之前做过了，但是由于思路不清晰，今天思路又卡了

比如：
         0
   1    2
3    4       5
   6       7

1->2
3->4->5
6->7

参考数据结构：
struct TREE_NODE
{
TREE_NODE(int n) : nVal(n), pLft(NULL), pRgt(NULL), pSib(NULL)
{}

int nVal;
TREE_NODE* pLft;
TREE_NODE* pRgt;
TREE_NODE* pSib;
};

对于每一个节点tnode，都要将其左右节点链接。如果没有右节点，则就要找和tnode同一层的节点，直到找到一个节点有左孩子或者右孩子，返回这个节点的孩子。

有一个问题要解决，在找和tnode同层的节点的孩子节点，必须保证tnode同层的所有节点已经链接了，用先左再右的递归不行，所以要用从右到左的递归。

TREE_NODE *find_next(TREE_NODE *root)
{
        while(root = root->pSib) {
                if(root->pLft || root->pRgt)
                        return root->pLft?root->pLft:root->pRgt;
        }
        return NULL;
}
void LinkLevelNodes(TREE_NODE *root)
{
        if(!root) return ;
        if(root->pLft)
                root->pLft->pSib = root->pRgt?root->pRgt:find_next(root);
        if(root->pRgt)
                root->pRgt->pSib = find_next(root);
        LinkLevelNodes(root->pRgt);
        LinkLevelNodes(root->pLft);
}