JZ36 二叉搜索树与双向链表

题目来源：牛客网

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

 注意:

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

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

 

 

代码实现：

class Solution {
public:
	void InOrder(TreeNode* cur,TreeNode*& prev)
	{
		if(cur==nullptr)
			return;
		InOrder(cur->left,prev);
		cur->left=prev;//当前节点的left指向前一个
		if(prev)//前一个的right指向当前节点
			prev->right=cur;
		prev=cur;
		InOrder(cur->right,prev);
	}
    TreeNode* Convert(TreeNode* pRootOfTree) {
        TreeNode* prev = nullptr;
		InOrder(pRootOfTree,prev);
		TreeNode* head = pRootOfTree;
		while(head && head->left)
		{
			head = head->left;
		}
		return head;
    }
};

思路：

我们对二叉树和链表分析，我们对二叉树走中序遍历的顺序就是链表里存的顺序

 因为题目要求空间复杂度的原因，我们要修改二叉树为链表

我们让二叉树的左指向前一个节点，right变为后一个节点即可

我们的prev使用了引用，原因是我们全程都只需要一个prev，否则会出现问题，prev是跟着cur变化的，大家可以画一画递归展开图进行理解

最后我们的头节点会变为链表中间的某一个节点，所以我们需要不断的找它的left，也就是前一个节点，直到找到头为止，然后返回即可