剑指offer笔记2：二维数组中的查找（Python）

剑指offer笔记2：二维数组中的查找（P44）

题目：在二维数组中查找指定数字

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

方法一：与右上角的数字比较

选取数组中右上角的数字。若该数字等于要查找的数字，则查找结束；若该数字大于要查找的数字，则删除该数字所在列；若该数字小于要查找的数字，则删除该数字所在行。在剩余区域继续寻找，直到找到要查找的数字或所剩区域为空。
python代码如下：

    def Find(self, target, array):
        # 选择右上角的数字进行比较，时间复杂度O(n)，空间复杂度O(1)
        if len(array) <= 0:
            # 若数组为空，则返回False
            return False
        row = 0
        # row为当前选取行，因为选取的是右上角的数字，所以行数从0开始
        rows = len(array) - 1
        # rows为数组的行数
        col = len(array[0]) - 1
        # col为数组的列数，也是当前选取的列
        while row <= rows and col >= 0:
            # 在缩小查找区域时，行数是+1的变化，列数是-1的变化
            if array[row][col] == target:
                # 若右上角的数字等于目标值，则查找成功，返回True
                return True
            elif array[row][col] > target:
                # 若右上角的数字大于目标值，则删除该列，即列数-1
                col -= 1
            else:
                # 若右上角的数字小于目标值，则删除该行，即行数-1
                row += 1
        return False

方法二：与左下角的数字比较

与方法一相同，可选取数组中左下角的数字。若该数字等于要查找的数字，则查找结束；若该数字小于要查找的数字，则删除该数字所在列；若该数字大于要查找的数字，则删除该数字所在行。在剩余区域继续寻找，直到找到要查找的数字或所剩区域为空。代码也只需要稍作修改：

    def Find(self, target, array):
        # 选择左下角的数字进行比较，时间复杂度O(n)，空间复杂度O(1)
        if len(array) <= 0:
            # 若数组为空，则返回False
            return False
        row = len(array) - 1
        # row为数组的行数，也是当前选取的行
        col = 0
        # col为当前选取列，因为选取的是左下角的数字，所以列数从0开始
        cols = len(array[0]) - 1
        # cols为数组的列数
        while col <= cols and row >= 0:
            print(array[row][col])
            # 在缩小查找区域时，行数是-1的变化，列数是+1的变化
            if array[row][col] == target:
                # 若左下角的数字等于目标值，则查找成功，返回True
                return True
            elif array[row][col] < target:
                # 若左下角的数字小于目标值，则删除该列，即列数+1
                col += 1
            else:
                # 若左下角的数字大于目标值，则删除该行，即行数-1
                row -= 1
        return False

小结

在解决二维数组的问题时，可以运用二维数组下标与偏移量的关系来查找我们想要的元素。