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

二维数组中的查找

题目描述

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

思路：

这是一个很特殊的二维数组，如下图所示，左上角a[0][0]的元素最小，右下角a[row][col]的元素最大。

那么我们依据这个特性，可以想到一种很简单的算法。从左下角a[0][col]这个元素开始遍历，把target和他进行比较，如果当前元素大于target元素，那么需要找到一个更小一些的元素，所以列数需要变小，即j--；如果当前元素小于target元素，需要找到一个更大的元素，行数变大，即i++。

在遍历数组中元素时，我们需要时刻注意是否越界。

代码：

public class Solution {
    public boolean Find(int target, int [][] array) {
        int m = array.length;
        int n = array[0].length;
        int i = m-1, j = 0;
        //判断当前位置是否越界
        while(i > 0  && j < n-1){
            //每次修改完当前元素的位置之后会继续判断，所以每次都需要判断是否越界
            if(target < array[i][j]&&i>0){
                i--;
            }
            if(target > array[i][j]&&j<n-1){
                j++;
            }
            if(target == array[i][j]){
                return true;
            }
        }
        return false;
    }
}

写代码过程中有两种具体实现方式，一种是上面这种修改了位置之后继续比较元素，所以每次比较元素大小时都要判断当前位置如果修改是否越界。还有一种是比较大小并修改位置之后就不继续比较，而是进入下一次循环。

public class Solution {
    public boolean Find(int target, int [][] array) {
        int m = array.length;
        int n = array[0].length;
        int i = m-1, j = 0;
        //判断当前位置是否越界，和上一种方法判断条件不同
        while(i >= 0  && j < n){
            //每次修改完当前元素的位置之后就进入下一次循环
            if(target < array[i][j]){
                i--;
            }
            else if(target > array[i][j]){
                j++;
            }
            else if(target == array[i][j]){
                return true;
            }
        }
        return false;
    }
}