题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
利用递归中序遍历实现转换,因为二叉搜索树中序遍历的结果是递增的,对于某个根节点,只需要先将左右子树转换为有序链表,再将链表连接起来即可。设置一个指针指示转换好的链表的最后一个节点,对于当前根节点,首先递归的转换左子树,然后将当前根节点的左指针指向转换好的链表当前的最后一个节点(当前最大的,同时比根节点小的节点),并且将最后一个节点的右指针指向当前根节点,然后将最后一个节点更新为当前根节点,递归转换右子树即可。
需要注意的是,当完成将最后一个节点的右指针指向当前根节点之前,必须确保最后一个节点不是空节点,因为空节点不能指向其他节点。
参考代码:
在线测试:
AC代码
空节点不能有指向?
/*
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==NULL)
return NULL;
TreeNode* pLastNode=NULL;
Covert(pRootOfTree, pLastNode);
TreeNode* pNode=pLastNode;
while(pNode->left)
{
pNode=pNode->left;
}
return pNode;
}
void Covert(TreeNode* pRootOfTree, TreeNode* &pLastNode)
{
if(pRootOfTree==NULL)
return;
if(pRootOfTree->left)
{
Covert(pRootOfTree->left, pLastNode);
}
pRootOfTree->left=pLastNode;
if(pLastNode)//确保最后一个节点不是空节点
{
pLastNode->right=pRootOfTree;
}
pLastNode=pRootOfTree;
if(pRootOfTree->right)
{
Covert(pRootOfTree->right, pLastNode);
}
}
};