LeetCode 116. 填充每个节点的下一个右侧节点指针。
给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
进阶:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
题解:
方法一:采用层次遍历。
/*
// 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;
Queue<Node> queue = new LinkedList<Node>();
queue.offer(root);
while(!queue.isEmpty()){
int size = queue.size();
for(int i = 0; i < size; i++){
Node top1 = queue.poll();
if(i == size - 1)
top1.next = null;
else
top1.next = queue.peek();
if(top1.left != null)
queue.offer(top1.left);
if(top1.right != null)
queue.offer(top1.right);
}
}
return root;
}
}
上述方法的时间复杂度为O(n),用队列保存节点,因此空间复杂度也为O(n).
方法二:可以将空间复杂度优化到O(1)。具体做法:
第一种情况为:连接同一父节点的两个子节点。假设node为父节点,则可以:
node.left.next = node.right;
第二种情况为:node和node1为相邻节点,node在左,即node.next = node1。这时连接下一层时,需要连接
node.right.next = node.next.left;
从上往下建立, 即在N-1层为N层建立连接。每次都找到当前层的最左侧节点,从左往右连接,根节点root不需要建立。
/*
// 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;
Node leftmost = root;
while(leftmost.left != null){
Node head = leftmost;
while(head != null){
//建立第一种连接
head.left.next = head.right;
//建立第二种连接
if(head.next != null)
head.right.next = head.next.left;
head = head.next;
}
leftmost = leftmost.left;
}
return root;
}
}
题目链接:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node
题解参考:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node/solution/tian-chong-mei-ge-jie-dian-de-xia-yi-ge-you-ce-2-4/