Leetcode_ 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

思路

怎样递归呢？能明显看出，变为线性后的序列是原树的先序遍历,直观感觉上，应该递归的方式应该和先序遍历的递归方式有很大的联系，先序线索化二叉树，更准确的说
我们这样来分息递归结构，每次访问一个节点的时候 ，我们必须要知道它的上一个访问的节点是什么，这样才能把上一个节点的右孩子指向当前访问的跟节点，所以我们用pre来保存，这样自己模拟的走两遍，就会发现递归结构和先序一样，但需要注意的是因为我们是把前驱节点的右孩子指向当前节点，所以 如果我们照搬先序遍历的话，当左子树访问完了，访问右子树指针已经变了，所以我们在进行右子树递归前应该用递归左子树前保存的指针，之前没有AC，后来看disscuss，发现自己没有把左子树赋值为NULL。

代码

class Solution {
public:
void flatten(TreeNode *root) {
TreeNode *p,*pre;
TreeNode *dump=new TreeNode(-1);
if (root==NULL) return;
flatternHelper(root,dump);
/*while(root!=NULL){
cout<<root->val<<endl;
root=root->right;
}*/
}
void flatternHelper(TreeNode *root,TreeNode* &pre)
{
if (root!=NULL)
{
//cout<<root->val<<endl;
pre->right = root;
pre->left = NULL;//this line can't be miss!
pre = root;
TreeNode *p = root->right;
flatternHelper(root->left,pre);
flatternHelper(p,pre);
}
}
};