BST转成双向链表

public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        // 根节点为空
        if(pRootOfTree == null) return null;

        if(pRootOfTree.left != null){
            // 先转换左子树,获得转换后的头指针
            TreeNode left = Convert(pRootOfTree.left);

            // 获得指向左子树的最后一个元素的指针
            while(left.right != null) left = left.right;

            // 与root链接,注意双向
            pRootOfTree.left = left;
            left.right = pRootOfTree;
        }

        if(pRootOfTree.right != null){
            // 同理,转换右子树
            TreeNode right = Convert(pRootOfTree.right);

            // 获得指向右子树的第一个元素的指针,这里直接就是right,与root双向链接
            pRootOfTree.right = right;
            right.left = pRootOfTree;
        }

        // 最后把pRootOfTree指向双向链表的第一个元素再返回
        while(pRootOfTree.left != null){
            pRootOfTree = pRootOfTree.left;
        }

        return pRootOfTree;
    }
}
全部评论

相关推荐

小叮当411:应该是1-3个月吧
点赞 评论 收藏
分享
07-04 09:21
已编辑
Java
推拿大师:这是hr发的钓鱼贴吗
投递字节跳动等公司8个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务