原创博文,转载请注明出处!
# 题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
- 二叉树节点的定义
struct TreeNode
{
int val;
struct TreeNode *left;
struct TreeNode *right;
}
- 二叉搜索树转双向链表的例子
# 思路
- 二叉搜索树的性质
- 二叉搜索树是左子树<根节点<右子树
- 二叉搜索树的中序遍历是递增的有序序列
- 二叉搜索树转双向链表的思路
- 首先:利用BST的中序遍历得到有序序列(递归)
- 其次:通过调整节点指针,将有序链表调整为双向链表
# 代码
class Solution {
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree == nullptr) return nullptr;
// 双向链表尾节点
TreeNode* list_last = nullptr;
// 递归建立双向链表
ConvertNode(pRootOfTree,list_last);
// 查找双向链表首节点
while(list_last->left != nullptr)
{
list_last = list_last->left;
}
// 返回双向链表的首节点
return list_last;
}
// 对BST中序遍历,得到有序序列;调整序列元素的指针,将有序序列调整为双向链表
void ConvertNode(TreeNode* cur,TreeNode *&list_last) // 注意形参
{
// 边界条件(递归出口)
if(cur==nullptr) return ;
// 遍历左子树
if(cur->left != nullptr) ConvertNode(cur->left,list_last);
// 建立双向链接
cur->left = list_last; // 单侧链接
if(list_last != nullptr) list_last->right = cur; // 单侧链接
list_last = cur;
//遍历右子树
if(cur->right != nullptr) ConvertNode(cur->right,list_last);
}
};