题目
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
思路
方法一:
将targrt与每行末尾比较,若比末尾小,则查看其是否在此行中
方法二:
阅读解答后,发现此题等于二分法
实现
方法一:
class Solution: def searchMatrix(self, matrix: List[List[int]], target: int) -> bool: if len(matrix) == 0 or len(matrix[0]) == 0: return False for row in range(len(matrix)): if target <= matrix[row][-1]: if target in matrix[row]: return True return False
方法二:
class Solution: def searchMatrix(self, matrix: List[List[int]], target: int) -> bool: row = len(matrix) if row == 0: return False col = len(matrix[0]) begin, last = 0,row*col -1 while begin <= last: mid = (begin + last) // 2 cur = matrix[mid//col][mid%col] if target == cur: return True elif target > cur: begin = mid + 1 else: last = mid -1 return False