首页 > 编程 > Python > 正文

Python实现二维有序数组查找的方法

2020-01-04 17:30:45
字体:
来源:转载
供稿:网友
这篇文章主要介绍了Python实现二维有序数组查找的方法,结合实例形式分析了Python二维有序数组的查找思路及具体实现技巧,需要的朋友可以参考下
 

本文实例讲述了Python实现二维有序数组查找的方法。分享给大家供大家参考,具体如下:

题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

这题目属于比较简单但又很不容易想到的,问了两个同学,大家一时都没有想出来怎么解决比较快。第一反应都是二分查找。对于每一行进行二分查找,然后查找过程可以把某些列排除掉,这是大家都能想到的基本的思路。

比较好的另一种思路是,首先选取数组右上角的数字,如果该数字等于要查找的数字,则查找结束;如果该数字大于要查找的数字,剔除这个数字所在的列,如果该数字小于要查找的数字,剔除这个数字所在的行。这样每一步都可以剔除一行或一列,查找的速度比较快。

python实现的代码:

# -*- coding:utf-8 -*-'''题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。'''def search(array, num):  # 参数合法性判断忽略  i = 0  j = len(array[0]) - 1  max_i = len(array) - 1  while i <= max_i and j >= 0:    if array[i][j] == num:      return True    elif array[i][j] > num:      j = j - 1    else:      i = i + 1  return Falseif __name__ == '__main__':  a = [[1, 2, 8, 9],     [2, 4, 9, 12],     [4, 7, 10, 13],     [6, 8, 11, 15],     ]  print search(a, 14)  print search(a, 7)  print search(a, 0)

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表