剑指offer之面试题36：二叉搜索树与双向链表

最新推荐文章于 2024-09-17 14:30:13 发布

yudidina

最新推荐文章于 2024-09-17 14:30:13 发布

阅读量77

点赞数

分类专栏：剑指offer 文章标签：链表指针数据结构二叉树算法

本文链接：https://blog.csdn.net/qq_43468570/article/details/105492692

版权

剑指offer 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

二叉搜索树与双向链表

1、题目

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建新的节点，只能调整树中节点指针的指向，二叉树的节点定义如下：

struct BinaryTreeNode
{
    int             m_nValue;
    BinaryTreeNode* m_pLeft;
    BinaryTreeNode* m_pRight;
};

输入参数：指向二叉搜索树根节点的指针

输出结果：双向链表的头节点

2、解题

解这道题的关键在于运用中序遍历二叉搜索树的思想。

有一个前提我们需把握住：中序遍历二叉搜索树，节点是递增排好序的。

我们可以假设现在处于根节点的位置，左子树已经排序完毕，那么此时双向链表的最后一个节点是左子树中值最大的节点，因为该二叉树是二叉搜索树，左子树的节点值都小于根节点，所以就可以把根节点加入链表，并让链表的尾指针指向根节点。接着继续去遍历转换右子树，并把根节点和右子树中最小的节点链接起来即可。

遍历完二叉搜索树后得到的是链表的尾指针，需要向前寻找到链表的头指针才算完成任务。

3、代码

BinaryTreeNode* Convert(BinaryTreeNode* pRoot) {
    BinaryTreeNode* pLastNodeInList = nullptr;
    ConvertNode(pRoot, &pLastNodeInList);

    BinaryTreeNode* pHeadOfList = pLastNodeInList;
    //需要考虑链表为空的情况，从尾节点开始找到头节点
    while (pHeadOfList != nullptr && pHeadOfList->m_pLeft != nullptr)
        pHeadOfList = pHeadOfList->m_pLeft;

    return pHeadOfList;
}

void ConvertNode(BinaryTreeNode* pRoot, BinaryTreeNode** pLastNodeInList) {
	if (pRoot == nullptr)
		return;
	//假入左子树不为空，左子树进行转换
	if (pRoot->m_pLeft != nullptr)
		ConvertNode(pRoot->m_pLeft, pLastNodeInList);

	pRoot->m_pLeft = *pLastNodeInList;
	if (*pLastNodeInList != nullptr)
		(*pLastNodeInList)->m_pRight = pRoot;

	*pLastNodeInList = pRoot;

	//假如右子树不为空，右子树进行转换
	if (pRoot->m_pRight != nullptr)
		ConvertNode(pRoot->m_pRight, pLastNodeInList);
}