给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点
https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node/
- 你只能使用常量级额外空间。
- 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
示例1:
提示:
Java解法
package sj.shimmer.algorithm.m3_2021;
import java.util.ArrayDeque;
import java.util.Queue;
/**
* Created by SJ on 2021/3/25.
*/
class D57 {
public static void main(String[] args) {
}
public Node connect(Node root) {
Queue<Node> queue = new ArrayDeque<>();
if (root != null) {
queue.add(root);
}
while (!queue.isEmpty()) {
Queue<Node> temp = new ArrayDeque<>();
Node tempNode = null;
while (!queue.isEmpty()) {
Node poll = queue.poll();
if (tempNode == null) {
tempNode = poll;
} else {
tempNode.next = poll;
tempNode = poll;
}
if (poll.left != null) {
temp.add(poll.left);
}
if (poll.right != null) {
temp.add(poll.right);
}
}
queue = temp;
}
return root;
}
public class Node {
public int val;
public Node left;
public Node right;
public Node next;
public Node() {
}
}
}
官方解
-
层次遍历
- 时间复杂度:O(N)
- 空间复杂度:O(N)
-
使用已建立的 next 指针