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;
		}
    }
};