剑指offer二维数组中的查找

题目描述 

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

思路

     二维数组内部是排好序的，每行从左到右，每列从上到下元素依次递增。

1. 首先拿目标值与矩阵中的右上角的元素去比较
2. 比右上角元素大（说明已经比这行中最大的数都大，再在这行中找没意义），删掉当前行（行数+1）
3. 比右上角元素小（说明已经比这列中最小的数都小，再在这列中找没意义），删掉当前列（列数-1）
4. 不断从矩阵的右上角开始切分，直到找到该元素(True)，或者矩阵已经被切分没了（False）

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        rows=len(array) #矩阵的行数
        cols=len(array[0]) #矩阵的列数
        row=0
        col=len(array[0])-1
        #判断二维矩阵循环结束的条件
        #while row>=0 and col>=0 and row<rows and col<cols:
        while row<rows and col>=0: 
            if array[row][col]<target:
                row+=1
            elif array[row][col]>target:
                col-=1
            else:
                return True
        return False

测试用例

if __name__=='__main__':
    s=Solution()
    array=[[0,1,2,3,4],
           [20,21,22,23,24],
           [30,31,32,33,34],
           [40,41,42,43,44]]
    print(s.Find(32,array))
    print(s.Find(29,array))