剑指offer - 二维数组中的查找

 二维数组中的查找

题目概要：

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

其实这个思路是讨论中提到最多的一种思路，并不是自己想出来的。

考虑到从左到右的递增和从上到下的递增顺序，我们先考虑左下角，左下角是最左列最后一行的数，他是这一列中最大的数，是这一行中最小的数，也就是说对这个数向上数字递减，向右数字递增。因此先用 target跟数组左下角进行比较，如果target比他小就往上找，如果比他大就往右找，以此类推，若找到相等的数返回true，否则返回false。

代码如下：

c++实现：

由于是vector，所以用size来确定行数和列数。运行时间：12ms

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size();
        int cols = array[0].size();
        int i = rows-1,j=0;
        while (i>=0 && j<cols){
            if (target<array[i][j])
                i--;
            else if (target > array[i][j])
                j++;
            else
                return true;
        }
        return false;
    }
};

python实现：

python是用len来求出行列数的，array[0]就是取出第一行的向量，再对其取长度就是列数。运行时间：342ms

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