要求
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。 每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true
示例 2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出:false
核心代码
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
m = len(matrix)
if m == 0:
return False
n = len(matrix[0])
if n == 0:
return False
x,y = m - 1,0
while(1):
if x < 0 or x >=m or y < 0 or y >= n:
break
val = matrix[x][y]
if val == target:
return True
elif val > target:
x -= 1
elif val < target:
y += 1
return False
解题思路:从左下角开始走,当前值比target大,就往上走;当前值比target小,就往右走;当前值=target就返回True。