[leetcode]114. Flatten Binary Tree to Linked List

题目链接：https://leetcode.com/problems/flatten-binary-tree-to-linked-list/#/description

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

         1
        / \
       2   5
      / \   \
     3   4   6

The flattened tree should look like:

   1
    \
     2
      \
       3
        \
         4
          \
           5
            \
             6

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        vector<TreeNode*> l;
        preorderTraversal(root, l);
        int n = l.size();
        for (int i = 1; i < n; i++) {
            TreeNode *prev = l[i - 1], *curr = l[i];
            prev->left = nullptr;
            prev->right = curr;
        }
    }

    void preorderTraversal(TreeNode* root, vector<TreeNode*> &l) {
        if (root != NULL) {
            l.push_back(root);
            preorderTraversal(root->left, l);
            preorderTraversal(root->right, l);
        }
    }
};

思路：

上面是题目及提示信息，大概意思是要按照前序遍历的顺序，将其变成一个但链表（但还是用树结构存），OK，开始分析下：

按照题目要求，它要把左子树的所有节点按照前序的顺序，挂载在右子树的链表中，那么我们首先遇到的第一个问题，把左子树挂载在右边，那之前的跟节点的右子树怎么处理？

上面这个问题，可以这样思考，我们假设已经把左子树挂载在右边，先忽视之前根节点的右子树，那么现在需要把其加载进去，那显然按照前序遍历的思想，我们肯定是把右子树挂载在左子树前序遍历的最后一个元素中，OK，这是我们目前的想法。

再思考下，我们只是简单的将根节点的左子树挂载在它和它的右子树中间，并没有将其“捋平”，这样就涉及到递归的处理，反复的将其，加进去。那这样就涉及到刚开始的一个问题，我们是否真的需要把右子树搁在左子树前序遍历的最后一个元素中？如下图所示，这里显然，我们只需要将4 搁在2的右子树位置上，（第二张图），然后再递归的捋顺根节点的right即可

             1                                           1

       2        4                                             2

    3        5     6                                    3        4

                                                                 5     6

class Solution {
public:
    void flatten(TreeNode* root) {
        TreeNode* now = root;
        while (now)
        {
            if (now->left)
            {
                TreeNode* pre = now->left;
                while (pre->right)
                {
                    pre = pre->right;
                }
                //将当前节点的右孩子连接到当前节点左孩子的的最右孩子
                pre->right = now->right;

                //当前节点的左孩子变为右孩子，左孩子置为NULL
                now->right = now->left;
                now->left = NULL;
            }
            now = now->right;
        }
    }
};