本文实例讲述了Python实现二维有序数组查找的方法。分享给大家供大家参考,具体如下:
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
这题目属于比较简单但又很不容易想到的,问了两个同学,大家一时都没有想出来怎么解决比较快。第一反应都是二分查找。对于每一行进行二分查找,然后查找过程可以把某些列排除掉,这是大家都能想到的基本的思路。
比较好的另一种思路是,首先选取数组右上角的数字,如果该数字等于要查找的数字,则查找结束;如果该数字大于要查找的数字,剔除这个数字所在的列,如果该数字小于要查找的数字,剔除这个数字所在的行。这样每一步都可以剔除一行或一列,查找的速度比较快。
python实现的代码:
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
|
# -*- 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程序设计有所帮助。