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 False
if __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)

更多关于Python相关内容感兴趣的读者可查看本站专题：《Python数据结构与算法教程》、《Python Socket编程技巧总结》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》

希望本文所述对大家Python程序设计有所帮助。

您可能感兴趣的文章:

• Python 检查数组元素是否存在类似PHP isset()方法
• python获取元素在数组中索引号的方法
• python简单获取数组元素个数的方法
• python实现判断数组是否包含指定元素的方法
• Python打印输出数组中全部元素
• Python简单计算数组元素平均值的方法示例
• Python实现返回数组中第i小元素的方法示例
• Python 比较两个数组的元素的异同方法
• Python实现变量数值交换及判断数组是否含有某个元素的方法
• Python实现找出数组中第2大数字的方法示例