题目描述
继续思考"填充每个节点指向最右节点的next指针" 这道题 如果给定的树可以是任意的二叉树呢?你之前的给出的算法还有效吗? 注意:
你只能使用常量的额外内存空间
/**
* Definition for binary tree with next pointer.
* struct TreeLinkNode {
* int val;
* TreeLinkNode *left, *right, *next;
* TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
* };
*/
class Solution {
public:
void connect(TreeLinkNode *root) {
while(root != NULL){
// 每次初始一个下一层的最左节点的虚拟头结点
TreeLinkNode TempFirst(0);
TreeLinkNode *temp = &TempFirst;
while(root){
if(root->left){
// 头结点把下一层的next连起来
temp->next = root->left;
temp = temp->next;
}
if(root->right){
temp->next = root->right;
temp = temp->next;
}
// 继续该层的遍历
root = root->next;
}
// 开始下一层
root = TempFirst.next;
}
}
};
错因分析
- 首先在初始化虚拟头结点时,指针的初始化,写错了
- 然后是,在进行下一层遍历时,没有结束本次的循环
- 最后就是一直想改进的问题,由于我每次都是在第一层循环里创建虚拟头结点,这样就造成了一定空间的浪费,但是将这条语句放到循环外的时候总是出现运行超时,是因为我没有注意到最后一行遍历的时候,root总是指向最后一行的最左节点了,并且与其相同了,导致循环出不去了。