Populating Next Right Pointers in Each Node ---LeetCode

https://leetcode.com/problems/populating-next-right-pointers-in-each-node/

解题思路:

观察可以发现,每一层从左至右每一个节点依次指向它右边的节点。因此可以用层序遍历 Binary Tree Level Order Traversal 来解决这道题,在遍历到这一层的时候,判断它是不是本层最右的元素,如果不是就指向旁边的元素,如果是就指向 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) {
        if (root == null) return ;

        Queue<TreeLinkNode> nodeQ = new LinkedList<>();
        Queue<Integer> depthQ = new LinkedList<>();

        if (root != null) {
            nodeQ.offer(root);
            depthQ.offer(1);
        }

        while (!nodeQ.isEmpty()) {
            TreeLinkNode node = nodeQ.poll();
            int depth = depthQ.poll();

            if (depthQ.isEmpty()) {
                node.next = null;
            } else if (depthQ.peek() > depth) {
                node.next = null;
            } else {
                node.next = nodeQ.peek();
            }

            if (node.left != null) {
                nodeQ.offer(node.left);
                depthQ.offer(depth + 1);
            }
            if (node.right != null) {
                nodeQ.offer(node.right);
                depthQ.offer(depth + 1);
            }
        }
    }
}

另外一种在 Program Creek 里学到的解法:可以不用队列。仅仅维护四个指针,其中两个指针指向两层的第一个节点,另外两个节点用于遍历层。

/**
 * 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) {
        if (root == null) return ;

        TreeLinkNode lastHead = root; // previous level's head
        TreeLinkNode lastCurr = null; // previous level's pointer
        TreeLinkNode currHead = null; // current level's head
        TreeLinkNode current  = null; // current level's pointer

        while (lastHead != null) {
            lastCurr = lastHead;

            while (lastCurr != null) {
                if (currHead == null) {
                    currHead = lastCurr.left;
                    current  = lastCurr.left;
                } else {
                    current.next = lastCurr.left;
                    current = current.next;
                }

                if (currHead != null) {
                    current.next = lastCurr.right;
                    current = current.next;
                }
                lastCurr = lastCurr.next;
            }
            lastHead = currHead;
            currHead = null;
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
根据引用\[1\]、\[2\]和\[3\]的代码,这是一个解决LeetCode 116题的Python代码。这个问题是关于填充每个节点的下一个右侧节点指针,使其指向其右侧节点的题目。代码中定义了一个Node类,其中包含了节点的值、左子节点、右子节点和下一个右侧节点指针。代码中的Solution类包含了一个connect方法,用于连接每个节点的下一个右侧节点指针。具体的实现方式有几种不同的方法,包括使用队列层序遍历、递归和带记忆的前序遍历。你可以根据自己的喜好选择其中一种方法来解决这个问题。 #### 引用[.reference_title] - *1* [【LeetCode】116. Populating Next Right Pointers in Each Node 解题报告(Python)](https://blog.csdn.net/L141210113/article/details/107134602)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [【LeetCode】116. 填充每个节点的下一个右侧节点指针 Populating Next Right Pointers in Each Node 解题...](https://blog.csdn.net/fuxuemingzhu/article/details/79559645)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [LeetCode第116题—填充每个节点的下一个右侧节点指针—Python实现](https://blog.csdn.net/qq_16184125/article/details/117201474)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值