1. 程式人生 > 實用技巧 >leetcode刷題-74搜尋二維矩陣

leetcode刷題-74搜尋二維矩陣

題目

編寫一個高效的演算法來判斷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