二分查找(二)

【找到 K 个最接近的元素】

给定一个排序好的数组,两个整数 k 和 x,从数组中找到最靠近 x(两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样,优先选择数值较小的那个数。

示例 1:

输入: [1,2,3,4,5], k=4, x=3
输出: [1,2,3,4]

示例 2:

输入: [1,2,3,4,5], k=4, x=-1
输出: [1,2,3,4]

链接:https://leetcode-cn.com/problems/find-k-closest-elements

    def findClosestElements(self, arr, k, x):
        """
        :type arr: List[int]
        :type k: int
        :type x: int
        :rtype: List[int]
        """
        # 查找>=目标值
        def find_target(arr, target):
            left = 0
            right = len(arr) - 1
            while left <= right:
                mid = (left + right) // 2
                if target > arr[mid]:
                    left = mid + 1
                else:
                    right = mid - 1
            return left
    
        index = find_target(arr, x)
        left, right = index - 1, index
        res = []
       # 向两边查找
        for i in range(k):
            if left < 0:
                res.append(arr[right])
                right += 1
            elif right == len(arr):
                res.append(arr[left])
                left -= 1
            else:
                if x - arr[left] <= arr[right] - x: 
                    res.append(arr[left])
                    left -= 1
                else:
                    res.append(arr[right])
                    right += 1
           
        return sorted(res)        

【寻找旋转排序数组中的最小值】

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。请找出其中最小的元素。你可以假设数组中不存在重复元素。

示例 1:

输入: [3,4,5,1,2]
输出: 1


示例 2:

输入: [4,5,6,7,0,1,2]

输出: 0

链接:https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array

    def findMin(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return -1
            
        left = 0
        right = len(nums) - 1
        target = nums[-1]  # 数组最右端的数字为target
        while left <= right:    		
            mid = (left + right) // 2
            if target > nums[mid]:		
                right = mid - 1   # 注意这里,,倒序不同
            else:
                left = mid + 1
        
        if right < 0:
            return nums[left]
        if left == len(nums):
            return nums[right]
        
        return nums[left]

【寻找旋转排序数组中的最小值 II】

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。请找出其中最小的元素。注意数组中可能存在重复的元素。

示例 1:

输入: [1,3,5]
输出: 1
示例 2:

输入: [2,2,2,0,1]
输出: 0

链接:https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array-ii

    if not nums:
        return -1

    left = 0
    right = len(nums) - 1
    while left <= right:
        mid = (left + right) // 2
        if nums[mid] < nums[right]:    # 一定在mid或mid左边
            right = mid
        elif nums[mid] > nums[right]:  # 一定在mid右边
            left = mid + 1
        else:
            right -= 1  # 相等时,一定mid或mid左边

    if right < 0:
        return nums[left]
    if left == len(nums):
        return nums[right]

    return nums[left]

  

https://leetcode-cn.com/problems/search-in-rotated-sorted-array/

https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/

https://leetcode-cn.com/problems/search-a-2d-matrix/

https://leetcode-cn.com/problems/search-a-2d-matrix-ii/

猜你喜欢

转载自www.cnblogs.com/rnanprince/p/11761940.html