python實現(xiàn)楊氏矩陣查找
本文實例為大家分享了python實現(xiàn)楊氏矩陣查找的具體代碼,供大家參考,具體內(nèi)容如下
問題描述:
在一個m行n列二維數(shù)組中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數(shù),輸入這樣的一個二維數(shù)組和一個整數(shù),判斷數(shù)組中是否含有該整數(shù)。有則返回1,無則返回0,輸入錯誤返回input error
問題分析:根據(jù)楊氏矩陣的規(guī)律可知,左上角是最小元素,右下角是最大元素,進行比較一次只能排除一個。而利用左下角或者右上角對比,一次可以排除一行或一列。
AC代碼以右上角為例:
- 當右上角大于要查找的數(shù)字時,排除一行;
- 當右上角大于要查找的數(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)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
對python打亂數(shù)據(jù)集中X,y標簽對的方法詳解
今天就為大家分享一篇對python打亂數(shù)據(jù)集中X,y標簽對的方法詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
python?tkinter庫的Text記錄點擊路經(jīng)和刪除記錄詳情
這篇文章主要介紹了python?tkinter庫的Text記錄點擊路經(jīng)和刪除記錄詳情,文章圍繞主題展開詳細的內(nèi)容介紹,具有一定的參考價值,感興趣的小伙伴可以參考一下2022-06-06
PyQt5執(zhí)行耗時操作導致界面卡死或未響應的原因及解決辦法
這篇文章主要給大家介紹了關(guān)于PyQt5執(zhí)行耗時操作導致界面卡死或未響應的原因及解決辦法,由于耗時的操作會獨占系統(tǒng)cpu資源,讓界面卡死在那里,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-12-12
pandas實現(xiàn)按照多列排序-ascending
這篇文章主要介紹了pandas實現(xiàn)按照多列排序-ascending,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-05-05
Python+Selenium+PIL+Tesseract自動識別驗證碼進行一鍵登錄
本篇文章主要介紹了Python+Selenium+PIL+Tesseract自動識別驗證碼進行一鍵登錄,具有一定的參考價值,有興趣的可以了解下2017-09-09

