题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路
先对二叉搜索树进行中序遍历,可获得递增的有序序列
其次,通过调整节点指针,将有序链表调整为双向链表
代码如下
class Solution {
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
if (pRootOfTree == nullptr) return nullptr;
//双向链表尾结点
TreeNode *last = nullptr;
//递归建立双向链表
Convertnode(pRootOfTree, last);
//查找双向链表的首结点
while (last -> left != nullptr)
last = last -> left;
//返回双向链表的首结点
return last;
}
// 对BST中序遍历,得到有序序列;调整序列元素的指针,将有序序列调整为双向链表
void Convertnode(TreeNode *cur, TreeNode *&last)
{
//边界条件(递归出口)
if (cur == nullptr) return;
//遍历左子树
if (cur -> left != nullptr) Convertnode(cur -> left, last);
//建立双向链接
cur -> left = last; //单侧链接
if (last != nullptr) last -> right = cur; //单侧链接
last = cur;
//遍历右子树
if (cur -> right != nullptr) Convertnode(cur -> right, last);
}
};
参考
https://www.cnblogs.com/wanglei5205/p/8780086.html