题目来源:
LeetCode题目:74. 搜索二维矩阵 - 力扣(LeetCode)
解题思路:
两次二分。第一次二分寻找第一列中小于等于target的最大值,第二次 在该行中找到小于等于target的最大值,最后判断是否存在该值及该值是否是 target。
解题代码:
#python3
class Solution:
def getIndex(matrix:List[int],target:int)->int:
left=0
right=len(matrix)-1
while(left<right):
mid=(left+right)//2+1
if matrix[mid]==target:
return mid
elif matrix[mid]<target:
left=mid
else:
right=mid-1
return left
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
n0=[]
for i in range(len(matrix)):
n0.append(matrix[i][0])
row=Solution.getIndex(n0,target)
if row<0 or row>=len(matrix):
return False
column=Solution.getIndex(matrix[row],target)
if column<0 or column>=len(matrix[0]):
return False
return matrix[row][column]==target
总结:
官方题解给出了两种解法。第一个是两次二分。第二个是一次二分,讲二维数组拼接为一维数组后再进行二分查找。