LeetCode114—Flatten Binary Tree to Linked List

原题

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

分析

最朴素的方法就是先序遍历把节点保存，然后在顺序构造不含左子树的树，网上看到有其他的方法，等学习研究了再更新。

代码

class Solution {private:    void dfs(TreeNode*root, vector<TreeNode*>&Nodes)    {        if (root == NULL)            return;        Nodes.push_back(root);        dfs(root->left, Nodes);        dfs(root->right, Nodes);    }public:    void flatten(TreeNode* root) {        if (root == NULL)            return;        vector<TreeNode*>Nodes;        dfs(root, Nodes);        int i;        root->left = NULL;        for (i = 0; i < Nodes.size() - 1; i++)        {            Nodes[i]->right = Nodes[i + 1];            Nodes[i]->left = NULL;        }        Nodes[i]->right = NULL;        root = Nodes[0];    }};