剑指Offer 36. 二叉搜索树展开为【双向】链表(Medium)/ LeetCode 109. 有序链表转换为【平衡】二叉搜索树(Medium)/ 升序数组转化为【平衡】二叉搜索树

  1. 二叉搜索树与双向链表(中序遍历,清晰图解)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

"""
# Definition for a Node.
class Node:
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
"""
class Solution:
    ### 1122 中序遍历(40 ms,14.7 MB)
    def treeToDoublyList(self, root: 'Node') -> 'Node':
        # 定义中序遍历递归函数dfs
        def dfs(cur):
            # 当前节点为空直接返回
            if not cur: return 

            # 中序遍历先递归左子树
            dfs(cur.left)

            # 若前一节点存在(即非头节点),则确定前一节点pre 和 当前节点cur的指向关系
            if self.pre:
                self.pre.right, cur.left = cur, self.pre
            # 若找到头节点,则头节点为当前节点cur,也即最左的叶子节点(此else语句只会执行一次)
            else:
                self.head = cur

            # 在递归右子树之前,需保存前一节点pre为当前节点cur,以便后续确定前、后节点的指向关系
            self.pre = cur 

            # 中序遍历最后递归右子树
            dfs(cur.right)

        # 若原始树为空,则直接返回
        if not root: return

        # 初始化前一节点pre为None(开始时前一节点pre并不存在)
        self.pre = None
        # 开始整棵树的中序遍历,这是为了将这颗树排序,让每一个节点以升序来排序
        dfs(root)

        # 中序遍历完成后,为了实现循环链表,需要连接头尾节点
        self.head.left, self.pre.right = self.pre, self.head

        return self.head # 最后返回头节点head

LeetCode 109. 有序链表转换为【平衡】二叉搜索树

在这里插入图片描述

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sortedListToBST(self, head: ListNode) -> TreeNode:
        def getLength(head: ListNode) -> int:
            res = 0
            while head:
                head = head.next
                res += 1
            return res
        
        # 中序遍历
        def buildTree(left: int, right: int) -> TreeNode:
            if left > right:
                return None

            mid = (left + right + 1) // 2 # 加1是因为这里是索引相加
            root = TreeNode()
            root.left = buildTree(left, mid - 1) # 先左

			# 第一次执行这部分代码是当上面的buildTree(left, mid - 1)已经遍历到最左叶节点
            # 这样就与中序遍历对应上,每遍历一个节点对应数组中的一个值,则无需单独寻找中位数
            nonlocal head
            root.val = head.val
            head = head.next

            root.right = buildTree(mid + 1, right) # 后右
            return root
        
        length = getLength(head)
        return buildTree(0, length - 1)

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值