链接:
题目:
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。 每行的第一个整数大于前一行的最后一个整数。
示例 1:输入: matrix = [ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ] target = 3 输出: true
示例 2:输入: matrix = [ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ] target = 13 输出: false
我的解法:
先找到这个数可能在的那行,再二分查找,找到了return True,找不到return False
class Solution(object):
def searchMatrix(self, matrix, target):
if len(matrix)==0 or len(matrix[0])==0:
return False
m = False
for i in range(len(matrix)):
if matrix[i][-1] >= target:
num = i
m = True
break
if m:
start = 0
end = len(matrix[0])
while start <= end:
mid = (start + end)/2
if matrix[num][mid] < target:
start = mid + 1
elif matrix[num][mid] > target:
end = mid - 1
else:
return True
return False
别人的解法:把所有的数抽成一行,二分查找。实现的时候是定了mid后,在原矩阵中找mid对应的位置
class Solution:
def searchMatrix(self, matrix, target):
if not matrix or target is None:
return False
rows, cols = len(matrix), len(matrix[0])
low, high = 0, rows * cols - 1
while low <= high:
mid = (low + high) / 2
# 根据mid在原矩阵找
num = matrix[mid / cols][mid % cols]
if num == target:
return True
elif num < target:
low = mid + 1
else:
high = mid - 1
return False