原题链接:https://leetcode-cn.com/problems/search-a-2d-matrix-ii/
编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
解题思路:
- 因为从左往右、从上到下都是升序;
- 从左下角向右上角走,当matrix[i][j]小于target,向上走一步,当matrix[i][j]大于target,向右走一步;
- 相等则返回True,若遍历到右上角都不相等,则返回False。
Python3代码:
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
row = len(matrix)-1
col = 0
while row >= 0 and col < len(matrix[0]):
if matrix[row][col] == target:
return True
elif matrix[row][col] < target:
col += 1
else:
row -= 1
return False