题目描述: 编写一个高效算法来判断 MxN 矩阵中,是否存在一个目标值。该矩阵具有以下性质:
- 每行中的整数从左至右按升序排列
- 每行的第一个整数大于前一行的最后一个整数
示例 :
输入: matrix = [[1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50]], target = 3
输出: true
使用 Python 使用 in 语句可以很容易实现这个任务,但很显然不是这道题的意义。其实很明显这个矩阵是排好序了的,所以理所当然的想到二分查找,而且是原地二分查找,使用而外空间把它降维再二分就会超时。
- 这里用到的技巧和有效数独里用到的是一样的,就是通过计算得到索引值,有效数独里是计算得到九个子块的编号,这里是计算矩阵元素的索引。我感觉这种方法在跟矩阵有关的题目中好像很有效的样子。代码如下:
def searchMatrix(matrix, target):
m = len(matrix)
if m == 0:
return False
n = len(matrix[0])
left = 0
right = m*n-1
while left<=right:
mid = (left+right) // 2
temp = matrix[mid // n][mid % n]
if temp == target:
return True
elif temp < target:
left = mid + 1
else:
right = mid - 1
return False