[剑指Offer]36-二叉搜索树与双向链表

链接

https://www.nowcoder.com/practice/947f6eb80d944a84850b0538bf0ec3a5?tpId=13&tqId=11179&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题意

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

解题思路

中序遍历,
若左子树不为空,找到!左子树转换成的链表的最后一个节点,并与根结点双向相连,
若右子树不为空,找到!右子树转换成的链表的首节点,并与根结点双向相连。
若左子树不空,返回左子树转成链表的首节点,否则,返回根节点。=》目标是返回整个链表的头部。

采用递归写法。

注意判空!
情景:当要访问一个节点的左孩子或右孩子时,当前节点是非空的吗?

其他知识

当要修改原指针指向的内存地址时,可函数传参 为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编辑  收藏  举报

导航