题目描述
思路分析
二叉树+BFS思想
实际上为
B
F
S
BFS
BFS思想,逐层添加
n
e
x
t
next
next指针,但因为存在上一层的
n
e
x
t
next
next指针,故可以省掉
B
F
S
BFS
BFS的栈
代码实现
/*
// 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) {
if(!root) return root;
auto begin=root;
while(root->left){
for(auto p=root;p;p=p->next){
p->left->next=p->right;
if(p->next) p->right->next=p->next->left;
}
root=root->left;
}
return begin;
}
};