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

Hints:

If you notice carefully in the flattened tree, each node's right child points to the next node of a pre-order traversal.

题目：将一棵二叉树，装换成一个单链表。

分析：hints中告诉我们，树的右指针做了链表中的next指针，并且是先序遍历的顺序。因此先序遍历这棵二叉树，分递归和迭代方法。注意，迭代的时候，要保存树的右孩子，防止右孩子丢失。

代码：

递归方法：

/** * Definition for binary tree * 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) return ;        TreeNode *keepRight = root->right;//保存右孩子                if(previsit!=NULL){            previsit->left=NULL;            previsit->right=root;//会修改右孩子        }        previsit=root;//注意更新                flatten(root->left);        flatten(keepRight);    }private:    TreeNode *previsit=NULL;//记录前一个访问的节点};

迭代方法：

class Solution {public:    //preorder 非递归方法    void flatten(TreeNode *root) {        if(!root) return ;        stack<TreeNode *> st;//辅助栈        st.push(root);        TreeNode *pre=0;//保存前一个节点                while(!st.empty()){            TreeNode *p=st.top();            st.pop();            if(p->right) st.push(p->right); //先右，栈的先进后出特性            if(p->left) st.push(p->left);                        if(pre){                pre->right=p;                pre->left=NULL;            }            pre=p;         }        pre->left=0;//尾节点置0        pre->right=0;            }};