leetcode（填充每个节点的下一个右侧节点指针 II）

songx_992025-09-27 20:44

给定一个二叉树：

复制代码

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL 。

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

思路：

采用层级遍历，也可以称广度优先遍历（BFS），建立一个队列存储，按每一层搜索查看

首先建立数组，将根节点放入，开始操作

计算此节点算在的这一层元素个数（队列长度），

对每一层，设一个pre节点初始化为null

for循环重复直到这一层元素均被遍历，使下一层它们的左右子结点均被加入。具体操作是

弹出首个节点，判断pre是否为null。如果为null表明此元素是这一层的首个元素，不需要指向。否则，设置他的next。判断完成后，将他的左右子结点（非空）加入队列。

java 复制代码

/*
// Definition for a Node.
class Node {
    public int val;
    public Node left;
    public Node right;
    public Node next;

    public Node() {}
    
    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, Node _left, Node _right, Node _next) {
        val = _val;
        left = _left;
        right = _right;
        next = _next;
    }
};
*/
class Solution {
    public Node connect(Node root) {
        if(root==null)
            return null;
        LinkedList<Node> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            int levelnum=queue.size();
            Node prev=null;
            for(int i=0;i<levelnum;i++) {
            Node curr = queue.poll();
            if(prev!=null) {
                prev.next = curr;
            }
                prev=curr;
            if (curr!=null&&curr.left != null) {
                    queue.add(curr.left);
            }
            if (curr!=null&&curr.right != null) {
                    queue.add(curr.right);
            }
            }
        }
        return root;
    }
}