题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。比如输入下图中左边的二叉搜索树,则输出转换之后的排序双向链表。
按照中序遍历的顺序,当我们遍历转换到根结点(值为10的结点)时,它的左子树已经转换成一个排序的链表了,并且处在链表中的最后一个结点是当前值最大的结点。我们把值为8的结点和根结点链接起来,此时链表中的最后一个结点就是10了。接着我们去遍历转换右子树,并把根结点和右子树中最小的结点链接起来。
很明显,转换它的左子树和右子树,由于遍历和转换过程是一样的,很自然地想到可以用递归。
对树采用中序遍历,并且利用一个pLastNode记录上一个结点,具体看程序,不难实现。
/*
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* pLastNode = nullptr;
ConvertNode(pRootOfTree,pLastNode);
TreeNode* res = pRootOfTree;
while(res->left != nullptr)
res = res->left;
return res;
}
void ConvertNode(TreeNode* pNode,TreeNode *&pLastNode)
{
if(pNode == nullptr)
return ;
ConvertNode(pNode->left,pLastNode);
pNode->left = pLastNode;
if(pLastNode != nullptr)
pLastNode->right = pNode;
pLastNode = pNode;
ConvertNode(pNode->right,pLastNode);
}
};