二维有序数组的查找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)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输...
    913c9536e19a阅读 409评论 0 0
  • 题目描述:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数...
    levinhax阅读 330评论 0 1
  • 数组的相关算法要简单一些,之前写过的和现在遇到的整理了一下。 数组:数组是较为简单的数据结构,它占据一块连续的内存...
    zero_sr阅读 1,360评论 0 2
  • 题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输...
    minningl阅读 471评论 0 0
  • 要求:不仅仅能实现相应的功能,还需要保证代码的鲁棒性,并且能够分析代码的空间复杂度和时间复杂度。 二维数组中的查找...
    二十四桥客_阅读 984评论 0 1