python实现杨氏矩阵查找_Python

python实现杨氏矩阵查找

2021-06-03 00:41ZzzMxin Python

这篇文章主要为大家详细介绍了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<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())