Populating Next Right Pointers in Each Node -- LeetCode

最新推荐文章于 2019-06-02 14:25:44 发布

iteye_18800

最新推荐文章于 2019-06-02 14:25:44 发布

阅读量212

点赞数

文章标签：数据结构与算法

本文介绍了一种特殊的二叉树层序遍历方法，通过将每一层节点组织为链表的方式进行遍历，避免使用额外的队列结构，实现了O(1)的空间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接: http://oj.leetcode.com/problems/populating-next-right-pointers-in-each-node/
这道题是要将一棵树的每一层维护成一个链表，树本身是给定的。其实思路上很接近层序遍历 Binary Tree Level Order Traversal ，只是这里不需要额外维护一个队列。因为这里每一层我们会维护成一个链表，这个链表其实就是每层起始的那个队列，因此我们只需要维护一个链表的起始指针就可以依次得到整个队列了。接下来就是有左加左入链表，有右加右入链表，知道链表没有元素了说明到达最底层了。算法的复杂度仍然是对每个结点访问一次，所以是O(n)，而空间上因为不需要额外空间来存储队列了，所以是O(1)。代码如下：

public void connect(TreeLinkNode root) {
    if(root == null)
        return;
    TreeLinkNode lastHead = root;
    TreeLinkNode pre = null;
    TreeLinkNode curHead = null;
    while(lastHead!=null)
    {
        TreeLinkNode lastCur = lastHead;
        while(lastCur != null)
        {
            if(lastCur.left!=null)
            {
                if(curHead == null)
                {
                    curHead = lastCur.left;
                    pre = curHead;
                }
                else
                {
                    pre.next = lastCur.left;
                    pre = pre.next;
                }
            }
            if(lastCur.right!=null)
            {
                if(curHead == null)
                {
                    curHead = lastCur.right;
                    pre = curHead;
                }
                else
                {
                    pre.next = lastCur.right;
                    pre = pre.next;
                }
            }                
            lastCur = lastCur.next;

        }
        lastHead = curHead;
        curHead = null;
    }
}

这道题是树的层序遍历 Binary Tree Level Order Traversal 的扩展，操作上会更加繁琐一些，因为是通过维护层链表来完成遍历，不过本质上还是一次广度优先搜索。