题目描述:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。比如输入下图中左边的二叉搜索树,则输出转换之后的排序双向链表。
解题思路:
1. 首先,我们知道:在二叉树中,每个结点都有两个指向子结点的指针。在双向链表中,每个结点也有两个指针,它们分别指向前一个结点和后一个结点。
2. 其次,由于要求转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点,这是因为中序遍历算法的特点是按照从小到大的顺序遍历二叉树的每一个结点。
3. 定义两个头结点,一个指向双向链表头,另一个将来指向链表尾,第一次运行程序时都初始化为二叉树根节点。
4. 随后的递归过程中,将右边头结点与当前结点相互指向,然后移动右边头结点。
Demo:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
TreeNode* pLeftHead = nullptr;
TreeNode* pRightHead = nullptr;
TreeNode* Convert(TreeNode* pRootOfTree)
{
// 递归调用叶子节点的左右节点返回nullptr
if(pRootOfTree == nullptr) return nullptr;
// 第一次运行时,它会使最左边叶子节点为链表第一个节点
Convert(pRootOfTree->left);
// 第一次运行时左边头结点和右边头结点重合,都是二叉树最左边的结点
if (pRightHead == nullptr)
pLeftHead = pRightHead = pRootOfTree;
else
{
// 定义新的指向
pRightHead->right = pRootOfTree;
pRootOfTree->left = pRightHead;
// 移动右边头结点(这一步很关键)
pRightHead = pRootOfTree;
}
//把右边叶子节点也插入到双向链表(rightHead已确定,直接插入)
Convert(pRootOfTree->right);
return pLeftHead;
}
};