剑指offer之二维数组中的查找（Java实现）

二维数组中的查找

NowCoder

题目描述

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
Consider the following matrix:
[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]
Given target = 5, return true.
Given target = 20, return false.

解题思路

//解法一：把每一行看成有序递增的数组，利用二分查找，通过遍历每一行得到答案，时间复杂度是nlogn
public class Solution {
    public boolean Find(int [][] array,int target) {
        for(int i=0;i<array.length;i++){
            int low=0;
            int high=array[i].length-1;
            while(low<=high){
                int mid=(low+high)/2;
                if(target>array[i][mid])
                    low=mid+1;
                else if(target<array[i][mid])
                    high=mid-1;
                else
                    return true;
            }
        }
        return false;
    }
}

/*
解法二：利用二维数组由上到下，由左到右递增的规律，那么选取右上角或者左下角的元素a[row][col]与target进行比较，当target小于元素
a[row][col]时，那么target必定在元素a所在行的左边,即col--；当target大于元素a[row][col]时，
那么target必定在元素a所在列的下边,即row++；
*/
public  class Solution{
    public boolean Find(int target, int[][] matrix) {
       if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
           return false;
       int rows = matrix.length, cols = matrix[0].length;
       int r = 0, c = cols - 1; // 从右上角开始
       while (r <= rows - 1 && c >= 0) {
           if (target == matrix[r][c])
               return true;
           else if (target > matrix[r][c])
               r++;
           else
               c--;
       }
       return false;
    }
}