牛客网剑指Offer 第一题：二维数组中的查找 python

题目

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

思路

从右上角的数出发，同行的数均小于该数，同列的数均大于该数。若该数小于target，则删除这一列，若该数大于target，则删除这一行。再取剩下数组中右上角的数。直到找到目标数字，返回True。若最后一个数还未找到，则返回False

代码

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        rowNum = len(array)
        colNum = len(array[0])
        if rowNum > 0 and colNum > 0:
            row = 0
            col = colNum-1
            while row < rowNum and col >=0 :
                if array[row][col] == target:
                    return True
                elif array[row][col] > target:
                    col -= 1
                else:
                    row += 1
        return False