给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {int val;Node *left;Node *right;Node *next; }
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
。
初始状态下,所有 next 指针都被设置为 NULL
。
示例 1:
输入:root = [1,2,3,4,5,6,7] 输出:[1,#,2,3,#,4,5,6,7,#] 解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化的输出按层序遍历排列,同一层节点由 next 指针连接,'#' 标志着每一层的结束。
示例 2:
输入:root = [] 输出:[]
/*
// Definition for a Node.
class Node {
public:int val;Node* left;Node* right;Node* next;Node() : val(0), left(NULL), right(NULL), next(NULL) {}Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}Node(int _val, Node* _left, Node* _right, Node* _next): val(_val), left(_left), right(_right), next(_next) {}
};
*/class Solution {
public:Node* connect(Node* root) {queue<Node*> que;if(root != NULL) que.push(root);while(!que.empty()){int size = que.size();Node* node; // 用来记录当前遍历到的结点Node* nodepre; // 用来记录当前结点的前一个结点for(int i = 0; i < size; i++){if(i == 0){node = que.front(); que.pop();nodepre = node; // 记录当前层的头结点}else{node = que.front();que.pop();nodepre->next = node; // 使前一个结点指向当前结点nodepre = node; // nodepre记录当前节点作为下次循环结点的前一个结点}if(node->left) que.push(node->left);if(node->right) que.push(node->right);}nodepre->next = NULL;}return root;}
};