LeetCode算法题：填充每个节点的下一个右侧节点指针connect

最新推荐文章于 2024-08-21 13:24:33 发布

「已注销」

最新推荐文章于 2024-08-21 13:24:33 发布

阅读量330

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/weixin_43777983/article/details/104855528

本文探讨了在完美二叉树中填充Next指针的问题，提出了两种解决方案：一种是使用队列进行层次遍历，另一种是采用递归方式实现节点间的连接。文章详细解释了每种方法的实现过程及优缺点。

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

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

示例：

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

不符合题目条件的通用做法，利用队列这种数据结构层次遍历树，定义一个先前节点pre，每遍历一个节点就让pre指向它，每一层结束后让pre指向空，然后遍历下一层。
题目要求常量空间，这种解法利用了空间复杂度为N的队列，所以不满足要求。

public Node connect(Node root) {
        if(root == null)
            return root;
        Queue<Node> q = new LinkedList<>();
        q.offer(root);
        while(!q.isEmpty()) {
            Node pre = q.peek();
            int count = q.size();
            while(count-- > 0) {
                Node cur = q.poll();
                if(cur.left != null)
                    q.offer(cur.left);
                if(cur.right != null)
                    q.offer(cur.right);
                pre.next = cur;
                pre = cur;
            }
            pre.next = null;
        }
        return root;
    }

第二种方法是用到了递归，
在这里插入图片描述
每当遇见一个节点时，就将它下一层的结点连接
最主要的是如何进行5 -> 6的连接，
那么就先判断目前的节点有没有next，如果有，则将下一层该节点的右子节点指向该节点next的左子节点。
最后递归的对每层每个节点进行连接。

public Node connect(Node root) {
        if(root == null)
            return null;
        Node left = root.left;
        Node right = root.right;
        if(left != null)
            left.next = right;
        Node next = root.next;
        if(next != null && right != null) {
            right.next = next.left;
        }
        connect(root.left);
        connect(root.right);
        return root;
    }