Lintcode---将二叉树拆成链表

将一棵二叉树按照前序遍历拆解成为一个

假链表

。所谓的假链表是说，用二叉树的 right 指针，来表示链表中的 next 指针。

注意事项

不要忘记将左儿子标记为 null，否则你可能会得到空间溢出或是时间溢出。

您在真实的面试中是否遇到过这个题？

Yes

样例

1
\
1          2
/ \          \
2   5    =>    3
/ \   \          \
3   4   6          4
\
5
\
6

思路：采用递归的方法，按照先序遍历，将节点存放起来，放在一个容器中；

然后新建一个头结点指针，遍历容器中的节点，依次将节点连接成一个链表。

但是，要注意将左儿子标记为NULL；

/**
* Definition of TreeNode:
* class TreeNode {
* public:
*     int val;
*     TreeNode *left, *right;
*     TreeNode(int val) {
*         this->val = val;
*         this->left = this->right = NULL;
*     }
* }
*/
class Solution {
public:
/**
* @param root: a TreeNode, the root of the binary tree
* @return: nothing
*/
/*
思路：采用递归的方法，按照先序遍历，先存放起来；
但是，要注意将左儿子标记为NULL；
*/

vector<TreeNode*> vec;
vector<TreeNode*> Node(TreeNode* root,vector<TreeNode*>& vec){
if(root==NULL){
return vec;
}
vec.push_back(root);
if(root->left!=NULL){
Node(root->left,vec);
}
if(root->right!=NULL){
Node(root->right,vec);
}
return vec;
}

void flatten(TreeNode *root) {
// write your code here
if(root==NULL){
return;
}

Node(root,vec);
root=new TreeNode();//创建新的头结点指针；
for(int i=0;i<vec.size();i++){
//多想想，第一次的写法为啥没有通过
vec[i]->left=NULL;
root->right=vec[i];
root=root->right;
}
}
};