二维数组中的查找--python实现

二维数组中的查找

题目描述:

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

解题思路:

由于原题说到,二维数组按行有序且按列有序,如果从矩阵的四个角出发,可以通过按行或者按列的方法找到目标值。本题中,我们以右上角为例,通过与目标值的比较,确定是向左移动还是向下移动。

解题代码:

class Solution:
    # array 二维列表
    def Find(self, target, array):
    
        #判断输入是否合法
        if not array:
            return None
    
        row_length = len(array)
        col_length = len(array[0])
        #将当前坐标位置设为右上角        
        cur_row = 0
        cur_col = col_length - 1

        #在合法边界内进行比较
        while (cur_row <= row_length-1 and cur_col >= 0):
            #如果与目标相等 返回
            if array[cur_row][cur_col] == target:
                return True
            #如果小于目标值,向下一行
            elif array[cur_row][cur_col] < target:
                cur_row += 1
            #如果大于目标值,向左一列
            else:
                cur_col -= 1
            

        return False

猜你喜欢

转载自www.cnblogs.com/ganshuoos/p/10897258.html