【剑指offer】二维数组中的查找

 

题目链接：二维数组中的查找

 

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

 

题解：这道题最容易想到的是o(n*m)的算法。[顺便BB一句，我参加17年的牛客的Google girl的面试时，面试官就出了这个题。] 回看一下题目的条件，就会发现这个数组是一个很特殊的数组。

1 2 3

4 5 6

7 8 9

取一个比较极端的例子，可以看出来，左上角永远是最小数，右下角永远是最大数。我们把目标值定为target，以副对角线划分，可以看出来小的在左边，大的在右边。也就是我们从右上角开始，如果target < a[i][j],那么j--,如果target > a[i][j],那么i++.这样出来的复杂度就o(m+n)了。

 

代码：

 

java

public class Solution {
    public boolean Find(int target, int [][] array) {
        int col = array[0].length-1;
        int row = 0;
        while(col>=0 && row < array.length){
            if(array[row][col] == target){
                return true;
            }
            else if(array[row][col] > target){
                col--;
            }
            else{
                row++;
            }
        }
        return false;
    }
}

 

c++

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