【leetcode】Flatten Binary Tree to Linked List

分析：

问题是将给定的二叉树变换成令一种形式，这样的类型的问题。其模式是，将左子树变换成某种形式，右子树也变换成这样的形式，然后再与根结点按规定的方式连接起来，那么总体就变换完毕了。

这个题我们就能够採用这样的形式。麻烦的地方就是在进行连接的时候。我们如果根为root，左子树变换后的根为root_left。右子树变换后的根为 root_right。那么连接的时候应该是root->right = root_left，再连接root_right的时候。应该是root_left所指向的一系列结点的最后一个结点和root_right连接。如果该结点为tail_left，那么就应该是tail_left->right = root_right。这样总体就连接好了。在详细连接的时候。可能要考虑某些结点为NULL的情况。

//

TreeNode *&link(TreeNode *&root)
	{
		if(root == NULL || root->left == NULL && root->right == NULL) 
			return root;
		TreeNode *root_left = NULL;
		TreeNode *root_right = NULL;
		root_left = link(root->left);
		root_right = link(root->right);
		if(root_left != NULL)
		{
			TreeNode *tail = root_left;
			//find the last one of root->left.
			while(tail->right) tail = tail->right;
			tail->right = root_right;
		}
		else
			root_left = root_right;
		root->right = root_left;
		root->left = NULL;
		return root;
	}
	void flatten(TreeNode *root) {
		if(root == NULL) return;
		root = link(root);
	}

我们再换个思路，题目中提示我们这样的形式事实上是树的先根遍历的结果。那么我们回忆下基本先根遍历是怎么处理的，我们都是简单的将根结点的val值进行打印。如今如果能够用额外的空间，那么我们就能够将先根遍历中的根结点放到一个顺序容器中。然后将每一个结点连起来就是题目要求的了。可是题目的意思非常明显不希望这样做，但我们还是希望用这样的形式，那怎么办呢？事实上我们不难发现，转换后的树形式上就是个单链表。那么问题就转换成单链表的建立。

void preVisit(TreeNode*& pre, TreeNode*root)
	{
		if(root == NULL)
			return;
		TreeNode *right = root->right;
		TreeNode *left = root->left;
		root->left = NULL;
		if(pre == NULL)
			pre = root;
		else
		{
			pre->right = root;
			pre = root;
		}
		if(left)
			preVisit(pre, left);
		if(right)
			preVisit(pre, right);
	}

上面的代码就是典型的先根遍历的模板。仅仅是模板中的万能visit函数做了一点略微复杂的事情。

void flatten(TreeNode* root)
	{
		if(root = NULL) return;
		TreeNode *pre = NULL;
		preVisit(pre, root);
	}

总结：还是重复前面文章中重复说过的。树的基本遍历的模式一定要熟记于胸，非常多衍生的问题都是能够套用模板的。