填充每个节点指向最右节点的next指针 ii

题目描述

继续思考"填充每个节点指向最右节点的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总是指向最后一行的最左节点了，并且与其相同了，导致循环出不去了。