剑指offer-二维数组中的查找-python

最新推荐文章于 2020-12-07 22:25:55 发布

亦万

最新推荐文章于 2020-12-07 22:25:55 发布

阅读量291

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 文章标签： python 数据结构算法剑指offer

本文链接：https://blog.youkuaiyun.com/GrinAndBearIt/article/details/87889855

剑指offer 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种在特定排序方式下的二维数组中查找整数的方法。通过从左下角开始，根据当前元素与目标数的关系，逐步缩小搜索范围，直至找到目标数或确定其不存在。此算法效率高，适用于按行和列递增排序的矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、题目描述

在一个二维数组中，每一行都按照从左往右递增以及每一列都按照从上往下递增的顺序排序。完成一个函数，输入这样的二维数组和一个整数，判断该整数是否在这个数组中

二、解题思路

书中的思路是从右上角不断往左下来剔除一行或者一列，我用的是从左下脚不断往右上提剔除一行或者一列，即当前位置的数如果大于num，则剔除当前行；如果当前位置的数小于num，则剔除当前列，需要考虑在两个边上的情况：如果在第一行的位置，如果当前位置的数比num大，则不存在相等，同理如果在最后一列，如果当前位置的数比num小，则不存在相等。

三、解题代码

def dimension_two_search(matrix, num):
    m, n = len(matrix), len(matrix[0])
    i, j = m-1, 0
    result = False
    while i >= 0 and j <= n-1:
        if matrix[i][j] == num:
            result = True
            break
        if i==0 and matrix[i][j] > num:
            break
        if j==n-1 and matrix[i][j] < num:
            break
        while matrix[i][j] > num and i > 0:
            i -= 1
        while matrix[i][j] < num and j < n-1:
            j += 1
    return result