leetCode 114. 二叉树展开为链表

题目描述：

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

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

 

思路：

根据例子可以看出这是一个前序遍历转化的链表，我们可以从根节点出发，先遍历左子树，

如果左子树不为空的话，那么就把左子树移到右子树上，让当前的的根节点的右指针指向这个左子树，再查找到这个左子树的最右子节点，把这个最右子节点的右指针，指向当前节点原来的右子树。最后把当前节点的左子树置为空。

然后再访问当前节点的右子树，递归的执行上述操作就可以。

代码如下：

/**
 * 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 preOrder(TreeNode* root){ 
        if(NULL != root->left){
            TreeNode* leftNode = root->left;
            TreeNode* right = root->right;
            root->left = NULL;
            root->right = leftNode;
            while(NULL != leftNode->right) leftNode = leftNode->right;
            leftNode->right = right;
            
            if(NULL == root->right)
                return;
            else
                preOrder(root->right);
        }
        
        if(NULL != root->right){
            preOrder(root->right);
        }
        
    }
    
    void flatten(TreeNode* root) {
        if(NULL == root)
            return;
        
        preOrder(root);
        return;
    }
};