python实现杨氏矩阵查找-创新互联
本文实例为大家分享了python实现杨氏矩阵查找的具体代码,供大家参考,具体内容如下

问题描述:
在一个m行n列二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。有则返回1,无则返回0,输入错误返回input error
问题分析:根据杨氏矩阵的规律可知,左上角是最小元素,右下角是大元素,进行比较一次只能排除一个。而利用左下角或者右上角对比,一次可以排除一行或一列。
AC代码以右上角为例:
- 当右上角大于要查找的数字时,排除一行;
- 当右上角大于要查找的数字时,排除一列;
- 相等则返回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):
# 列表模拟二维数组
p = list(map(int, sys.stdin.readline().split()))
arr.append(p)
# 获取要查找的数
s = int(sys.stdin.readline().strip())
i = 0
j = m - 1
while i=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())
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
网站栏目:python实现杨氏矩阵查找-创新互联
标题来源:http://www.jxjierui.cn/article/jsooh.html


咨询
建站咨询
