leetcode 将已排序的 数组/链表 转换为二叉搜索树(BST),Python实现

通过递归找到有序数组或链表的中点作为根节点,构建二叉搜索树。详细介绍了将有序序列转化为BST的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

思路:不论是数组还是链表,递归地找到他的root(即序列的中点),并返回。

1. 将数组转换为二叉树:

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    # @param num, a list of integers
    # @return a tree node
    # 12:37
    def sortedArrayToBST(self, num):
        if not num:
            return None
        
        mid = len(num)//2  #“//”表示整数除法;“/”浮点数除法;
        root = TreeNode(num[mid])
        left = num[:mid]
        right = num[mid+1:]
        root.left = self.sortedArrayToBST(left)
        root.right = self.sortedArrayToBST(right)
        return root

2. 将链表转换为二叉树:

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def sortedListToBST(self, head):
        if not head:
            return None
        if not head.next:
            return TreeNode(head.val)
            
        Lroot = self.find_root(head)
        Troot = TreeNode(Lroot.val)
        Troot.left = self.sortedListToBST(head)
        Troot.right = self.sortedListToBST(Lroot.next)
        return Troot
        
    def find_root(self,head):  #找到链表的中点作为后半部分的起点,并将链表切断。
        if not head or not head.next:
            return head
        
        slow = head
        fast = head
        pre = head
        while fast and fast.next:
            pre = slow
            slow = slow.next
            fast = fast.next.next
        
        pre.next = None
        return slow


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值