LeetCode114—Flatten Binary Tree to Linked List

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];
    }
};