【劍指 Offer】 04. 二維數(shù)組中的查找

題目描述
在一個(gè) n * m 的二維數(shù)組中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請(qǐng)完成一個(gè)函數(shù),輸入這樣的一個(gè)二維數(shù)組和一個(gè)整數(shù),判斷數(shù)組中是否含有該整數(shù)。
示例:
現(xiàn)有矩陣 matrix 如下:

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]

給定 target = 5,返回 true。
給定 target = 20,返回 false。
限制:
0 <= n <= 1000
0 <= m <= 1000
解題思路
最容易想到的就是暴力遍歷矩陣,但是由于題目給出的矩陣具有一定規(guī)律性,所以在看了Leetcode上精選的題解后,還是挺巧妙的。首先從矩陣最左下角的元素a開始進(jìn)行判斷,當(dāng)taget<a時(shí),則忽略改行,往上遍歷;當(dāng)target>a時(shí),則忽略該列,往右遍歷。直到遍歷到與target值相等,輸出True。否則將整個(gè)矩陣都遍歷完成之后還是不能找到與target值相等的數(shù),則輸出False.

程序代碼

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        #len(matrix)是矩陣行數(shù),len(matrix[0])是矩陣列數(shù)
        i, j = len(matrix) - 1, 0 
        while i >= 0 and j < len(matrix[0]):
            if matrix[i][j] > target:
                i -= 1
            elif matrix[i][j] < target:
                j += 1
            else:
                return True
        
        return False

復(fù)雜度分析

  • 時(shí)間復(fù)雜度O(M+N):其中NM分別是矩陣行數(shù)和列數(shù)
  • 空間復(fù)雜度O(1):i,j指針使用常數(shù)大小額外空間
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容