leetcode刷題-74搜尋二維矩陣
阿新 • • 發佈:2020-08-08
題目
編寫一個高效的演算法來判斷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 Trueelif target > cur: begin = mid + 1 else: last = mid -1 return False