【找到 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/
原文:https://www.cnblogs.com/rnanprince/p/11761940.html