题目描述
给定一个二叉树填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。初始状态下，所有 next 指针都被设置为 NULL。
进阶：
- 你只能使用常量级额外空间。
- 使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。
题解
诶？和之前的题目还不一样了，不能直接算出来什么时候该分层了。那就，只能，递归了呗？（队列依然是可以使用的，以后再说吧）
核心是getNextNoNullChild，根据root，找到下一级右手第一个
然后分情况讨论，对每个节点：
左子右子都有，则左子next指向右子，右子next指向getNextNoNullChild
只有左子，左子指向getNextNoNullChild，
只有右子，右子指向getNextNoNullChild，
注意：递归时要先递归右子树，否则上级节点next关系没建好，下级无法成功getNextNoNullChild
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
| Node* getNextNoNullChild(Node* root){
Node* t = root;
while (t->next != NULL){
if (t->next->left!= NULL) {
return t->next->left;
}
if (t->next->right != NULL){
return t->next->right;
}
t = t->next;
}
return NULL;
}
Node* connect(Node* root) {
if (!root||(!root->left&&!root->right))return root;
if (root->left&&root->right){
root->left->next = root->right;
root->right->next = getNextNoNullChild(root);
}
else if (!root->right){
root->left->next = getNextNoNullChild(root);
}
else{
root->right->next = getNextNoNullChild(root);
}
root->right = connect(root->right);
root->left = connect(root->left);
return root;
}
|