溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本文實例為大家分享了python實現楊氏矩陣查找的具體代碼，供大家參考，具體內容如下
問題描述：
在一個m行n列二維數組中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序排序。請完成一個函數，輸入這樣的一個二維數組和一個整數，判斷數組中是否含有該整數。有則返回1，無則返回0，輸入錯誤返回input error
問題分析：根據楊氏矩陣的規律可知，左上角是最小元素，右下角是最大元素，進行比較一次只能排除一個。而利用左下角或者右上角對比，一次可以排除一行或一列。
AC代碼以右上角為例：
代碼如下：
# 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):
# 列表模擬二維數組
p = list(map(int, sys.stdin.readline().split()))
arr.append(p)
# 獲取要查找的數
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())
如有疑問，歡迎交流和指正。
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持億速云。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
