LeetCode117 填充每个节点的下一个右侧节点指针 II
方法一:层次遍历
对于每一层,我们都可以计算每一层的数量,在这一层中,我们进行next的赋值。
利用for循环就可以实现每一层的赋值
class Solution {
public Node connect(Node root) {
if (root == null) {
return null;
}
Queue<Node> queue = new LinkedList<Node>();
queue.offer(root);
while (!queue.isEmpty()) {
int n = queue.size();
Node last = null;
for (int i = 1; i <= n; ++i) {
Node f = queue.poll();
if (f.left != null) {
queue.offer(f.left);
}
if (f.right != null) {
queue.offer(f.right);
}
if (i != 1) {
last.next = f;
}
last = f;
}
}
return root;
}
}
方法二:利用上一层的next指针,进行下一层next指针的赋值,可以减少空间
根节点开始。因为第 0 层只有一个节点,不需要处理。
可以在上一层为下一层建立next 指针。该方法最重要的一点是:位于第 x层时为第 x + 1 层建立next 指针。
一旦完成这些连接操作,移至第 x + 1层为第 x + 2 层建立 next指针。
当遍历到某层节点时,该层节点的 next 指针已经建立。这样就不需要队列从而节省空间。
每次只要知道下一层的最左边的节点,就可以从该节点开始,像遍历链表一样遍历该层的所有节点。
class Solution {
Node last = null, nextStart = null;
public Node connect(Node root) {
if (root == null) {
return null;
}
Node start = root;
while (start != null) {
last = null;
nextStart = null;
for (Node p = start; p != null; p = p.next) {
if (p.left != null) {
handle(p.left);
}
if (p.right != null) {
handle(p.right);
}
}
start = nextStart;
}
return root;
}
public void handle(Node p) {
if (last != null) {
last.next = p;
}
if (nextStart == null) {
nextStart = p;
}
last = p;
}
}