题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路: 中序遍历二叉搜索树即为有序的序列,再递归建立双向链表即可。
C++
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
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);
}
};