在行列都排好序的矩阵中找数
生活随笔
收集整理的這篇文章主要介紹了
在行列都排好序的矩阵中找数
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目】 給定一個有N*M的整型矩陣matrix和一個整數K, matrix的每一行和每一 列都是排好序的。實現一個函數,判斷K 是否在matrix中。 例如: 0 1 2 5 2 3 4 7 4 4 4 8 5 7 7 9 如果K為7, 返回true;? 如果K為6,返 回false。
要求】 時間復雜度為O(N+M),額外空間復雜度為O(1)
思路:
1、從右上角的數開始尋找(row=0 col=M-1),比較當前數matrix[row][col]和k的關系
2、如果等于k,返回true表示已經找到
如果比k大,因為矩陣每一列都是排好序,所以在當前數所在的列中,處于當前數下方的數都比k大,則沒有必要繼續在第col列上尋找,令col = col- 1,重復步驟
如果比k小,因為矩陣每一列都是排好序,所以在當前數所在的行中,處于當前數左方的數都比k小,則沒有必要繼續在第row行上尋找,令row?= row +?1,重復步驟
3、如果找到越界都沒有發現與K相等的數,則返回false
def isContain(matrix,k):row = 0col = len(matrix[0])-1while row < len(matrix) and col > -1:if matrix[row][col] == k:return Trueelif matrix[row][col] > k:col -=1else:row +=1return FalseisContain([[0,1,2,5],[2,3,4,7],[4,4,4,8],[5,7,7,9]],7)?
總結
以上是生活随笔為你收集整理的在行列都排好序的矩阵中找数的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 反转单向和双向链表
- 下一篇: 打印两个有序链表的公共部分