🍕🎉🥓🎊🚗🐷😱❤😀❣🥙🎇🕵️♀️😎🎑🎁🎉😚🎇
题目:
- 给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
- 填充它的每个
next
指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将next
指针设置为NULL
。初始状态下,所有next
指针都被设置为NULL
。
示例:
- 输入: r o o t = [ 1 , 2 , 3 , 4 , 5 , 6 , 7 ] root = [1,2,3,4,5,6,7] root=[1,2,3,4,5,6,7]
- 输出: [ 1 , # , 2 , 3 , # , 4 , 5 , 6 , 7 , # ] [1,\#,2,3,\#,4,5,6,7,\#] [1,#,2,3,#,4,5,6,7,#]
- 解释: 给定二叉树如左图所示,你的函数应该填充它的每个
next
指针,以指向其下一个右侧节点,如右图所示。序列化的输出按层序遍历排列,同一层节点由next
指针连接,'#'
标志着每一层的结束。
解题思路:
利用队列,每遍历一个元素,就将其加入到队列中,判断当前元素是否为该层最后一个元素,若是的话则将其next
置为空,否则next
指向下一个元素。
C++版本
class Solution {
public:
Node* connect(Node* root) {
if(root == NULL) return NULL;
queue<Node*> q;
q.push(root);
while(!q.empty()){
int size = q.size();
for(int i = 0; i < size; i++){
Node* temp = q.front();
q.pop();
if(i != size - 1) temp->next = q.front();
else temp->next = NULL;
if(temp->left) q.push(temp->left);
if(temp->right) q.push(temp->right);
}
}
return root;
}
};
Java版本
class Solution {
public Node connect(Node root) {
Queue<Node> q = new LinkedList<>();
if(root == null) return null;
q.offer(root);
while(!q.isEmpty()){
int size = q.size();
for(int i = 0; i < size; i++){
Node temp = q.poll();
if(i == size - 1) temp.next = null;
else temp.next = q.peek();
if(temp.left != null) q.offer(temp.left);
if(temp.right != null) q.offer(temp.right);
}
}
return root;
}
}