Populating Next Right Pointers in Each Node II

最新推荐文章于 2024-08-25 16:11:27 发布

havedream_one

最新推荐文章于 2024-08-25 16:11:27 发布

阅读量3.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： LeetCode

本文链接：https://blog.csdn.net/havedream_one/article/details/42555293

LeetCode 专栏收录该内容

85 篇文章

订阅专栏

本文探讨了在非完全二叉树情况下，如何使用常量额外空间来填充每个节点的下一个指针，详细介绍了算法实现及其实例应用。

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

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

Note:

You may only use constant extra space.

For example,
Given the following binary tree,

After calling your function, the tree should look like:

         1 -> NULL
       /  \
      2 -> 3 -> NULL
     / \    \
    4-> 5 -> 7 -> NULL

/**
 * Definition for binary tree with next pointer.
 * public class TreeLinkNode {
 *     int val;
 *     TreeLinkNode left, right, next;
 *     TreeLinkNode(int x) { val = x; }
 * }
 */
public class Solution {
    public void connect(TreeLinkNode root) {
        Queue<TreeLinkNode> queue = new LinkedList<TreeLinkNode>();
        int level = 1;//记录需要出队的元素数，
        if(root == null ){
            return;
        }
        queue.offer(root);
        int count = 0;
        TreeLinkNode p = null;
        while(queue.size() > 0){
            int level1 = 0;//下一层需要出队的个数
            while(count < level - 1){
                p = queue.poll();
                p.next = queue.peek();
                if(p.left != null){
                    queue.offer(p.left);
                    level1++;
                }
                if(p.right != null){
                    level1++;
                    queue.offer(p.right);
                }
                count++;
            }
            p = queue.poll();
            if(p.left != null){
                queue.offer(p.left);
                level1++;
            }
            if(p.right != null){
                level1++;
                queue.offer(p.right);
            }
            level = level1;
            count = 0;
        }
    }
}

与上一题类似，只不过这次树不再是完全二叉树了，但是只要记录下需要出队的元素数，一切还是OK的！