原题如下:
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
这道题看上去挺简单,就是个先序遍历,但关键是在in-place的情况下进行指针的重定向,由于先序遍历的结果是根左右,所以针对每个节点,在转化成链表时,需要将其右子树挂在其左子树的后边(即将其右子树作为其左子树的右子树),然后将其左子树挂在其后(即将其左子树作为其右子树),同时将其左子树置空,所以针对当前节点需要按照右左根的顺序进行处理,而在遍历时只需保证根节点在最后处理(即在处理根节点时需保证其左右子树已经序列化,而其左右子树的序列化顺序不分先后),所以遍历既可按照左右根的顺序,也可按照右左根的顺序。
class Solution {
public:
void flatten(TreeNode *root) {
if(root == NULL)
return;
//printf("%d\n",root->val);
flatten(root->left);//1
flatten(root->right);//2,此两句的顺序可以调换
TreeNode *p = root;
if(p->left == NULL)
return ;
p = p->left;
while(p->right != NULL)
p = p->right;
p->right = root->right;
root->right = root->left;
root->left = NULL;
}
另外,在这道题中,如果不考虑in-place的要求,可以按照先序遍历存入vector中,然后再重新连接起来,这样方法虽然笨,但至少能够解决问题。