您好,登录后才能下订单哦!
密码登录
登录注册
点击 登录注册 即表示同意《亿速云用户服务条款》
本文实例为大家分享了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进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。