114. 二叉树展开为链表

二叉树展开为链表

原题：

给你二叉树的根结点 root ，请你将它展开为一个单链表：

• 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
• 展开后的单链表应该与二叉树 先序遍历 顺序相同。

示例 1：

输入：root = [1,2,5,3,4,null,6]
输出：[1,null,2,null,3,null,4,null,5,null,6]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [0]
输出：[0]

提示：

• 树中结点数在范围 [0, 2000] 内
• -100 <= Node.val <= 100

进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？

解题方法：

解法一：

class Solution {
public:
    void flatten(TreeNode* root) {
        if (root == nullptr) return;
        flatten(root->left);
        flatten(root->right);
        if (root->left != nullptr) {
            auto pre = root->left;
            while (pre->right != nullptr) pre = pre->right;
            pre->right = root->right;
            root->right = root->left;
            root->left = nullptr;
        }
        root = root->right;
        return;
    }
};

做题小结：

其实这个解法的思路跟我的差不多，即后序遍历，对最后遍历到的非叶节点，进行左节点的右节点等于右节点，然后右节点等于左节点的操作。但是问题就在于我把是否为叶结点单独拿出来判断并return了，但是其实是不用的，而且我少了一步在左节点的最右节点加上右节点部分的内容，相当于我只考虑了叶结点的情况。

还是太菜了呀！！