117. 填充每个节点的下一个右侧节点指针 II
给定一个二叉树填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
思路
层序遍历
一开始居然没有想到层序遍历,直接层序遍历,可以获得当前层所有节点,连线即可。
代码
/*
// 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 == nullptr) {
return nullptr;
}
queue<Node*> q;
q.push(root);
while(!q.empty()) {
int n = q.size();
//保存上一个节点
Node* last = nullptr;
for(int i = 0; i < n; i++) {
Node* tmp = q.front();
q.pop();
if(tmp -> left != nullptr) {
q.push(tmp -> left);
}
if(tmp -> right != nullptr) {
q.push(tmp -> right);
}
//访问的不是第一个节点
//连接
if(i != 0) {
last -> next = tmp;
}
last = tmp;
}
}
return root;
}
};
next节点
可以在访问第i层时,连接好第i+1层的节点信息,访问根节点时,就直接处理第二层的信息即可,这样在访问每一层时,就可以当作链表来访问以及处理下一层的next连接了。
代码
/*
// 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:
//负责连接,并处理下一层的起始点
void handle(Node* &last, Node* &p, Node* &nextStart) {
//last节点存储的是上一个节点信息
//last不为空,证明访问的不是层第一个节点
if(last != nullptr) {
last -> next = p;
}
//nextStart存储的是下一层的起始点,为空证明访问的是层第一个节点
if(nextStart == nullptr) {
nextStart = p;
}
//更新last节点
last = p;
}
Node* connect(Node* root) {
if(root == nullptr) {
return nullptr;
}
Node* tmp = root;
while(tmp) {
Node* last = nullptr, *nextStart = nullptr;
//把当前层当作链表访问
for(Node* p = tmp; p != nullptr; p = p -> next) {
//处理下一层
if(p -> left != nullptr) {
handle(last, p -> left, nextStart);
}
if(p -> right != nullptr) {
handle(last, p -> right, nextStart);
}
}
//到达下一层起始点
tmp = nextStart;
}
return root;
}
};