[剑指Offer]36-二叉搜索树与双向链表
链接
题意
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
解题思路
中序遍历,
若左子树不为空,找到!左子树转换成的链表的最后一个节点,并与根结点双向相连,
若右子树不为空,找到!右子树转换成的链表的首节点,并与根结点双向相连。
若左子树不空,返回左子树转成链表的首节点,否则,返回根节点。=》目标是返回整个链表的头部。
采用递归写法。
注意判空!
情景:当要访问一个节点的左孩子或右孩子时,当前节点是非空的吗?
其他知识
当要修改原指针指向的内存地址时,可函数传参 为TreeNode** pNode。
代码
class Solution {
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(!pRootOfTree){
return nullptr;
}
if(!pRootOfTree->left&&!pRootOfTree->right){
return pRootOfTree;
}
//找到左子树转换成的链表的最右节点,连上根节点
TreeNode* pLeftList=nullptr;
pLeftList=Convert(pRootOfTree->left);
if(pLeftList){//
TreeNode* p=pLeftList;
while(p->right){//
p=p->right;
}
pRootOfTree->left=p;
p->right=pRootOfTree;
}
//找到右子树转换成的链表的最左节点,连上根结点
TreeNode* pRightList=nullptr;
pRightList=Convert(pRootOfTree->right);
if(pRightList){
TreeNode* p=pRightList;
while(p->left){
p=p->left;
}
pRootOfTree->right=p;
p->left=pRootOfTree;
}
//返回链表
if(pLeftList){
return pLeftList;
}
else{
return pRootOfTree;
}
}
};
posted on 2019-03-22 11:04 coding_gaga 阅读(111) 评论(0) 编辑 收藏 举报