LeetCode 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

 

 

将一颗二叉树转换为链，要求原地进行。

可以观察到，需要的输出结果实际就是原树的先序遍历序列。

先序遍历，将新发现的节点连接到上一个节点的左节点，则：

1、如果新发现的节点是上一个节点的左子节点，则原本的连接关系不需要修改。

2、否则，上一个节点必然是叶节点，将新发现的节点连接为上一个节点的左子节点，

由于上一个节点已经探测结束，所以将新发现的节点连接在其左子树上不会影响后续的遍历。

最后递归扫描修改后的树的左子节点，将左节点连接为右子节点，将左子节点设为空即可。

 

代码：

/** * 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 to_left(TreeNode* root,TreeNode *&last)//当前位置，上一位置；先序遍历，将后一节点连在前一节点左子树上。{if(root==NULL)return;if(last!=NULL)last->left=root;last=root;//修改上一位置to_left(root->left,last);//递归to_left(root->right,last);}    void flatten(TreeNode *root) {TreeNode* last=NULL;to_left(root,last);while(root!=NULL)//递归将左子树移位右子树，清空右子树连接{root->right=root->left;root->left=NULL;root=root->right;}    }};