python

超轻量级php框架startmvc

python实现杨氏矩阵查找

更新时间:2020-06-26 20:12:01 作者:startmvc
本文实例为大家分享了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())

如有疑问,欢迎交流和指正。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

python杨氏矩阵查找 python杨氏矩阵 python矩阵查找