剑指offer——面试题4：二维数组中的查找

二维数组中的查找

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

• 解法
  思路： 该二维数组的有序的， 应充分利用其有序性。设该数组为m行n列，所查找的整数值为A。可知最后一列的数都是所在行的最大值，于是开始就依次与该列的值作比较， 过程如下：

  • 设i表示行号（i从0开始），j表示列号（j从n-1开始， 即最后一列）， 若A == array[i][j],那么就找到该值了，返回True； 若A>array[i][j], 说明A比第i行的所有值都要大，则该行舍弃，跳到下一行， 重新开始比较， i+=1； 若A< array[i][j], 说明第j列的所有值都比A大， 则该列舍弃，跳到前一列， j-=1， A只可能出现在在第i行和第j-1列相交的左下区域了，在该区域中重新按以上方法重新比较。
    时间复杂度为O(n)

• 代码

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