【Leetcode】【Medium】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

解题思路：

看上去很简单，使用二叉树的中序遍历就可以实现。但是题目的小曲点在于要在原tree上修改，如果将左儿子放在右儿子位置上，会丢失右子树，导致遍历失败。

解决方法有两种：

1、不使用多余空间，将右子树放在左子树中，最右边儿子的右儿子位置上，也正好满足了，leftchild -> data -> rightchild的中序顺序，但是缺点是要遍历多次树，时间复杂度高；

2、只关注左儿子，将左儿子放到右儿子的位置上，同时使用栈结构，将所有右子树挨个入栈。在没有左儿子时，取出栈中元素。

第二种解法的代码：

/**
 * 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) {
        stack<TreeNode*> rights;
        TreeNode* node = root;
        
        while (node != NULL) {
            while (node->left) {
                if (node->right)
                    rights.push(node->right);
                node->right = node->left;
                node->left = NULL;
                node = node->right;
            }
            
            if (node->right) {
                node = node->right;
                continue;
            }
            
            if (!rights.empty()) {
                node->right = rights.top();
                node = node->right;
                rights.pop();
            } else {
                break;
            }
        }
        
        return;
    }
};