114. Flatten Binary Tree to Linked List

114 . Flatten Binary Tree to Linked List

Difficulty: Medium

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

解题思路

仔细看看这道题会很快发现，我们其实只要先序遍历整棵树，并且在遍历过程当中将重排问题同步解决就可以了。

在实现过程中，最为麻烦的是将函数尽可能优化，因为要一边搜索一边重排，这就意味着如果没有合理的结构就会有不好的结果。

我在这里使用的就是循环思路，将左子树不断插入到右子树中去即可。中间就是不断遍历左边的树并且一步步让root指向root-〉right即可。

具体实现

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public:    void flatten(TreeNode* root) {        if(root == NULL) return;        while(root)        {            if(root->left)            {                TreeNode *pre = root->left;                while(pre->right)                    pre = pre->right;                pre->right = root->right;                root->right = root->left;                root->left = NULL;            }            root = root->right;        }    }};