二维数组中的查找——牛客网、剑指offer（python解决）

题目描述：

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

解法1：暴力解决

直接套两层循环，对二维数组进行遍历，二维数组中的元素与输入的整数进行一一比较，只要有相同的，即返回True

代码如下：

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        row = len(array)
        col = len(array[0]) 
        for i in range(row):
            for j in range(col):
                if target == array[i][j]:
                    return True
        return False

运行结果：

解法2：对角搜索法

根据题目中设定的二维数组的元素排序规律，可以从二维数组的右上角开始与target整数进行比较，假如target小于右上角的元素，最后一列就直接舍去，因为每列都是由小到大排列的；假如target大于右上角的元素，整行就舍去，直接从下一行的右上角开始寻找。

代码如下:

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

运行结果：

 

总结：可以从运行结果看出，第二种方法的运行时间会比第一种方法快，这也是优化了寻找相同整数的算法的结果。