二叉搜索树与双向链表

描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。如下图所示

数据范围：输入二叉树的节点数 0≤n≤10000≤n≤1000，二叉树中每个节点的值 0≤val≤10000≤val≤1000
要求：空间复杂度O(1)O(1)（即在原树上操作），时间复杂度 O(n)O(n)

注意:

1.要求不能创建任何新的结点，只能调整树中结点指针的指向。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode，有左右指针，其实可以看成一个双向链表的数据结构

4.你不用输出双向链表，程序会根据你的返回值自动打印输出

输入描述：

二叉树的根节点

返回值描述：

双向链表的其中一个头节点。

示例1

输入：

{10,6,14,4,8,12,16}

返回值：

From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;

说明：

输入题面图中二叉树，输出的时候将双向链表的头节点返回即可。     

示例2

输入：

{5,4,#,3,#,2,#,1}

返回值：

From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;

说明：

                    5
                  /
                4
              /
            3
          /
        2
      /
    1
树的形状如上图       

解析

1.首先保证有序，那就是中序遍历

2.为了进行连接，由于是单项箭头，所以必须要建立一个prev指针

核心要求：当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继

 

 为了连接prev和cur，prev->right = cur;是在下一层进行的。

prev = cur,修改一级指针，因此prev需要引用传参。

/*

不采用引用传参，真正出问题的prev = cur;是这个语句

这个语句的目的是更新 prev 指针，使其指向当前遍历到的节点 cur。然而，如果 prev 是通过值传递的（即传递了一个指针的副本），那么这个赋值操作只会改变函数内部 prev 副本的值，而不会影响到原始的 prev 指针。

修改prev->right ,只需要传入right的上一级指针：prev

如果想修改prev，那就得传入prev的上一级指针：prev的引用或者二级指针

*/

代码

class Solution {
public:
			//prev需要在另一层函数中进行修改，所以引用传参。
	void InorderConvert(TreeNode* cur, TreeNode*& prev)
	{
		if (cur == nullptr)
			return;

		InorderConvert(cur->left, prev);
		
		cur->left = prev;	//中序遍历到最小节点之后
		
		if (prev)	//下一层的prev才能链接到上一层的cur
			prev->right = cur;
		
		prev = cur;

		InorderConvert(cur->right, prev);
	}

    TreeNode* Convert(TreeNode* pRootOfTree) {

		TreeNode* cur = pRootOfTree, *prev = nullptr;

		InorderConvert(cur, prev);
		
		TreeNode* phead = pRootOfTree;
		while (phead && phead->left)	//找到头节点（不能为空）
		{
			phead = phead->left;
		}

		return phead;
    }
};