二叉树展开为链表【前序遍历】

给定一个二叉树，原地将它展开为一个单链表

例如，给定二叉树

1<br />   / \<br />  2   5<br /> / \   \<br />3   4   6<br />将其展开为：

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 preOrderTraversal(TreeNode *node, vector<TreeNode *> & pre_order_res){
        if (node == nullptr)
            return;
        pre_order_res.push_back(node);
        preOrderTraversal(node->left, pre_order_res);
        preOrderTraversal(node->right, pre_order_res);
    }
    // 收集所有前序遍历的结果，然后将其拼接起来
    void flatten(TreeNode* root) {
        vector<TreeNode *> pre_order_res;
        preOrderTraversal(root, pre_order_res);
        int node_size = pre_order_res.size();

        TreeNode * node = root;
        for(int i=1; i<node_size; ++i){
            node->left = nullptr;
            node->right = pre_order_res[i];
            node = node->right;
        }
    }
    // 时间复杂度： O(n);
    // 空间复杂度： O(n);
};

PS: 当然还有空间复杂度为O(1)的解法，这里为了不做介绍。