剑指offer : 二叉搜索树与双向链表

题目描述:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

思路:

首先,二叉搜索树的中序遍历就是有序的,因此我们借用中序遍历的概念

二叉搜索树的特点:  左子树的值 < 根节点的值 < 右子树的值

因此只要将左子树最右边的节点与根节点与右子树最左侧的节点连接就可以得到一个有序链表。

先对树进行遍历得到最左侧的节点,也就是整棵树最小的节点

再找右子树中最小的节点

将这两个节点与根节点连接起来

递归实现。

代码:

class Solution {
public:
	TreeNode * Convert(TreeNode* pRootOfTree)
	{
		if (pRootOfTree == nullptr)
		{
			return nullptr;
		}
		//pLastNode始终指向链表的最后一个节点
		TreeNode* pLastNode = nullptr;
		ConvertNode(pRootOfTree, &pLastNode);
		TreeNode* pHeadNode = pLastNode;
		//找到左子树转换后的最后一个节点
		while (pHeadNode != nullptr&&pHeadNode->left != nullptr)
		{
			pHeadNode = pHeadNode->left;
		}
		//我们需要返回头结点
		return pHeadNode; 
	}

	void ConvertNode(TreeNode* pRootNode, TreeNode** pLastNode)
	{
		if (pRootNode == nullptr)
		{
			return;
		}
		TreeNode* CurNode = pRootNode;
		if (CurNode->left != nullptr)
		{
			ConvertNode(CurNode->left, pLastNode);
		}
		CurNode->left = *pLastNode;
		if (*pLastNode != nullptr)
		{
			(*pLastNode)->right = CurNode;
		}
		//将当前节点保存为尾节点,接续遍历
		*pLastNode = CurNode;
		if (CurNode->right != nullptr)
		{
			ConvertNode(CurNode->right, pLastNode);
		}

	}
};

猜你喜欢

转载自blog.csdn.net/Shile975/article/details/89002504