python實(shí)現(xiàn)楊氏矩陣查找
本文實(shí)例為大家分享了python實(shí)現(xiàn)楊氏矩陣查找的具體代碼,供大家參考,具體內(nèi)容如下
問題描述:
在一個m行n列二維數(shù)組中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數(shù),輸入這樣的一個二維數(shù)組和一個整數(shù),判斷數(shù)組中是否含有該整數(shù)。有則返回1,無則返回0,輸入錯誤返回input error
問題分析:根據(jù)楊氏矩陣的規(guī)律可知,左上角是最小元素,右下角是最大元素,進(jìn)行比較一次只能排除一個。而利用左下角或者右上角對比,一次可以排除一行或一列。
AC代碼以右上角為例:
- 當(dāng)右上角大于要查找的數(shù)字時,排除一行;
- 當(dāng)右上角大于要查找的數(shù)字時,排除一列;
- 相等則返回1
- 全部查找失敗則返回0
代碼如下:
# coding=utf-8 import sys def solve(): try: # 獲取行(n)和列(m) a = sys.stdin.readline().split() n, m = int(a[0]), int(a[1]) arr = [] for i in range(n): # 列表模擬二維數(shù)組 p = list(map(int, sys.stdin.readline().split())) arr.append(p) # 獲取要查找的數(shù) s = int(sys.stdin.readline().strip()) i = 0 j = m - 1 while i<n and j>=0: if arr[i][j] == s: # 相等返回1查找成功 return 1 elif arr[i][j] < s: # 小于要查找的元素,行加1 i += 1 else: # 大于要查找的元素,列加1 j -= 1 return 0 except Exception: return "input error" if __name__ == "__main__": print(solve())
如有疑問,歡迎交流和指正。
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Python中類方法@classmethod和靜態(tài)方法@staticmethod解析
這篇文章主要介紹了Python中類方法@classmethod和靜態(tài)方法@staticmethod解析,python中存在三種方法,分別為常規(guī)方法(定義中傳入self)、@classmethod修飾的類方法、@staticmethod修飾的靜態(tài)方法,,需要的朋友可以參考下2023-08-08Python中使用裝飾器和元編程實(shí)現(xiàn)結(jié)構(gòu)體類實(shí)例
Python中使用裝飾器和元編程實(shí)現(xiàn)結(jié)構(gòu)體類實(shí)例,本文的方法算是一種Python的黑魔法技術(shù),并非Python的常規(guī)寫法,需要的朋友可以參考下2015-01-01Python辦公自動化之網(wǎng)絡(luò)監(jiān)控和壓縮文件處理
Python辦公?動化是利用Python編程語?來創(chuàng)建腳本和程序,以簡化、加速和?動化?常辦公任務(wù)和工作流程的過程,本文主要介紹了如何進(jìn)行網(wǎng)絡(luò)監(jiān)控和壓縮文件處理,感興趣的可以了解下2023-12-12已解決不小心卸載pip后怎么處理(重新安裝pip的兩種方式)
這篇文章主要介紹了已解決不小心卸載pip后怎么處理(重新安裝pip的兩種方式),本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-04-04Pandas操作兩個Excel實(shí)現(xiàn)數(shù)據(jù)對應(yīng)行的合并
本文主要介紹了Pandas操作兩個Excel實(shí)現(xiàn)數(shù)據(jù)對應(yīng)行的合并,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-01-01Python numpy生成矩陣、串聯(lián)矩陣代碼分享
這篇文章主要介紹了Python numpy生成矩陣、串聯(lián)矩陣代碼分享,具有一定參考價值,需要的朋友可以了解下。2017-12-12