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许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。