leetcode(填充每个节点的下一个右侧节点指针 II)
给定一个二叉树:
struct Node {int val;Node *left;Node *right;Node *next; }
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
。
初始状态下,所有 next 指针都被设置为 NULL
。
思路:
采用层级遍历,也可以称广度优先遍历(BFS),建立一个队列存储,按每一层搜索查看
首先建立数组,将根节点放入,开始操作
计算此节点算在的这一层元素个数(队列长度),
对每一层,设一个pre节点初始化为null
for循环重复直到这一层元素均被遍历,使下一层它们的左右子结点均被加入。具体操作是
弹出首个节点,判断pre是否为null。如果为null表明此元素是这一层的首个元素,不需要指向。否则,设置他的next。判断完成后,将他的左右子结点(非空)加入队列。
/*
// 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;}
}