JZ26 二叉搜索树与双向链表***

题目描述

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

思路

大体思路懂了,但是递归还是有点迷糊
通过画图我们可以知道,将二叉搜索树转换成双向链表,需要对根的左右指针进行操作,将根的左指针指向左子树的最右,将根的右指针指向右子树的最左
关键点:按照中序遍历的顺序(左->根->右)
所以当遍历到根的时候,左子树的最右就是当前链表的尾结点;而右子树的最左就是遍历右子树的第一个节点

代码

class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree==NULL)
            return NULL;
        TreeNode* pLastNodeInList=NULL;
        ConvertNode(pRootOfTree,pLastNodeInList);

        //pLastNodeInList指向双向链表的尾结点,需要返回头结点
        TreeNode* pHeadOfList=pLastNodeInList;
        while(pHeadOfList!=NULL&&pHeadOfList->left!=NULL)
            pHeadOfList=pHeadOfList->left;      //头结点就是最左

        return pHeadOfList;

    }
    //中序遍历:左——>根——>右
    void ConvertNode(TreeNode* pRoot,TreeNode*& pLastNodeInList)
    {
        if(pRoot==NULL)
            return;
        TreeNode* pCurrent=pRoot;

        if(pCurrent->left!=NULL)   //左节点
            ConvertNode(pCurrent->left,pLastNodeInList);  
        //遍历完左节点后,当前链表的最后一个节点就是左子树上的最右
        pCurrent->left=pLastNodeInList;  //把当前节点的左指针指向链表的最后一个元素
        if(pLastNodeInList!=NULL)        //遍历根节点(链表的指针移到根节点上去)
            pLastNodeInList->right=pCurrent;

        pLastNodeInList=pCurrent;   

        if(pCurrent->right!=NULL)   //遍历右子树
            ConvertNode(pCurrent->right,pLastNodeInList);
    }
};
全部评论

相关推荐

11-03 14:38
重庆大学 Java
AAA求offer教程:我手都抬起来了又揣裤兜了
点赞 评论 收藏
分享
去B座二楼砸水泥地:不过也可以理解,这种应该没参加过秋招
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务