题目要求
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。 输出链表的”头节点“。
解题分析
二叉搜索树和排序链表有什么关系?
二叉搜索树也是一种排序的数据结构,所有的左子节点的值都小于根结点的值,所有右子节点的值都大于根节点的值。那我们可以根据这个特性吧BST转换成排序的双向链表。我们只要把原先指向左孩子节点的指针调整为链表中指向前一个节点的指针,把原先指向右孩子节点的指针调整为链表中指向后一个节点的指针即可。
重点是如何进行转换呢?
由于需要转换后是排好顺序的,所以我们用中序遍历去遍历二叉排序树中的每个节点。具体的步骤结合下面的代码了解一下。
主要代码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* pre = nullptr;
ConvertHelper(pRootOfTree,pre);
TreeNode * res = pRootOfTree; // 排好序不行还要输出链表的头
while(res->left)
res = res->left; // 找到链表的头 即BST最左边的叶子结点(最小)
return res;
}
private:
void ConvertHelper(TreeNode* cur, TreeNode*& pre)
{
if(cur == nullptr)
return;
ConvertHelper(cur->left,pre); // 先对左子树进行转换
cur->left = pre; // 把根节点(cur)接到排好序的左子树上
if(pre) pre->right = cur; // 把根节点(cur)接到排好序的左子树上
pre = cur; // 移到根节点
ConvertHelper(cur->right, pre); // 对右子树进行转换
}
};