二维数组中的查找

最新推荐文章于 2024-08-30 12:08:57 发布

---Lavender---

最新推荐文章于 2024-08-30 12:08:57 发布

阅读量252

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer（Python版）文章标签： python

本文链接：https://blog.youkuaiyun.com/Tenderzhe/article/details/78394409

剑指offer（Python版）专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种在特殊排序的二维数组中查找特定整数的方法。通过从数组的右上角开始，逐步比较并调整搜索范围，实现了高效查找。文章提供了Python实现代码及测试示例。

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

题目描述：

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

答案：

答案思路：从数组的右上角开始，当target值大于对应值时，切换到下一行，当target小于对应值时，取值向左边移动，如此循环直至找到target

# -*- coding:utf-8 -*-
class Solution:
    def Find(self, target, array):
        if array == [] or array == [[]]:
            return False

        rawnum = len(array)
        print rawnum
        colnum = len(array[0])
        if type(target) == float and type(array[0][0]) == int:
            target = int(target)
        elif type(target) == int and type(array[0][0]) == float:
            target = float(int)
        elif type(target) != type(array[0][0]):
            return False

        i = colnum - 1
        j = 0
        while i >= 0 and j < rawnum:
            if array[j][i] < target:
                j += 1
            elif array[j][i] > target:
                i -= 1
            else:
                return True
        return False

简单测试：

import numpy as np
array = np.array(([1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]))
def find(array,rows,cols,key):
    found = False
    if array.size != 0 and rows > 0 and cols > 0:
        row = 0
        col = cols-1
        while row < rows and col >= 0:
            if key < array[row][col]:
                col-=1
            elif key > array[row][col]:
                row+=1
            else:
                found = True
                break
        return found


print(find(array,4,4,2))