您的位置:首页 > 脚本大全 > > 正文

怎么用python做随机矩阵(python实现杨氏矩阵查找)

更多 时间:2022-01-16 00:57:30 类别:脚本大全 浏览量:2680

怎么用python做随机矩阵

python实现杨氏矩阵查找

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

问题描述:

在一个m行n列二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。有则返回1,无则返回0,输入错误返回input error

问题分析:根据杨氏矩阵的规律可知,左上角是最小元素,右下角是最大元素,进行比较一次只能排除一个。而利用左下角或者右上角对比,一次可以排除一行或一列。

AC代码以右上角为例:

  • 当右上角大于要查找的数字时,排除一行;
  • 当右上角大于要查找的数字时,排除一列;
  • 相等则返回1
  • 全部查找失败则返回0

代码如下:

  • ?
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • # 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())
  • 如有疑问,欢迎交流和指正。

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

    原文链接:https://blog.csdn.net/qq_34178562/article/details/79648265

    您可能感兴趣