输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向
思路总结:本题是结合二叉搜索树与双向链表的考点
本质上是分为三个部分,左、右子树 和根节点 进行考虑,即将左子树转换成链表后,将尾节点与根节点连接,再将根节点带入递归表达式求右子树节点,最终实现转换,注意将尾节点移回到头部的节点
/*
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)
{
TreeNode* pLastNode = NULL;
ConvertNode(pRootOfTree,&pLastNode);
//pLastNode始终是双向链表的最后一个节点,
//通过递归调用后,将节点移动到头结点
TreeNode* pHeadNode = pLastNode;
while(pHeadNode != NULL &&pHeadNode->left != NULL)
pHeadNode = pHeadNode->left;
return pHeadNode;
}
//转换函数
void ConvertNode(TreeNode* pNode, TreeNode** pLastNode){
if(pNode == NULL) return ;
TreeNode *pCurrent = pNode;//获取当前节点
if(pCurrent->left != NULL) //左节点非空,就继续向下遍历
ConvertNode(pCurrent->left,pLastNode);
pCurrent->left = *pLastNode;//将上一个尾节点作为现在这个节点的做节点,相当于节点的连接
if(*pLastNode != NULL)
(*pLastNode)->right = pCurrent;//双向链表右指针
*pLastNode = pCurrent; //将当前节点设置为尾节点,继续遍历
if(pCurrent->right != NULL) //遍历右子树
ConvertNode(pCurrent->right,pLastNode) ;
}
};