剑指 Offer 04. 二维数组中的查找

题目

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

示例:

现有矩阵 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]
]
给定 target = 5，返回 true。

给定 target = 20，返回 false。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

1. 由于每个一维数组都是有序的，所以可以遍历每个一位数组，使用二分法查找target，时间复杂度为n*lgn。

   class Solution {
       public boolean findNumberIn2DArray(int[][] matrix, int target) {
           if (matrix.length == 0 || matrix[0].length == 0) {
               return false;
           }
           
           int arrayNum = matrix.length;
           int arrayLen = matrix[0].length;
           
           for (int i = 0; i < arrayNum; i++) {
               int min = matrix[i][0];
               int max = matrix[i][arrayLen - 1];
               
               if (target >= min && target <= max) {
                   boolean res = binarySearch(matrix, i, target);
                   if (res) {
                       return res;
                   } 
               }
           }
           
           return false;
       }
       
       boolean binarySearch(int[][] matrix, int num, int target) {
           int[] arr = matrix[num];
           int low = 0;
           int high = arr.length - 1;
           
           while (low <= high) {
               int mid = (low + high) / 2;
               
               if (arr[mid] == target) {
                   return true;
               }
               
               if (target < arr[mid]) {
                   high = mid - 1;
               } else {
                   low = mid + 1;
               }
           }
           
           return false;
       }
   }
   

2. 将题目中的条件全部用到。题目中说，从左到右是递增的，从上到下也是递增的，那么从下往上，从右往左都是递减的。

   所以我们可以选择左下角的元素，如果target大于左下角元素，那么索引向右移，如果小于，那么向上移动。

   或者可以选择右上角的元素，如果target大于右上角元素，那么索引可以向下移动，如果小于，那么向左移动。

   class Solution {
       public boolean findNumberIn2DArray(int[][] matrix, int target) {
           if (matrix.length == 0 || matrix[0].length == 0) {
               return false;
           }
           
           int i = matrix.length - 1, j = 0;
           
           while (i >= 0 && j <= matrix[0].length - 1) {
               if (matrix[i][j] == target) {
                   return true;
               }
               
               if (matrix[i][j] < target) {
                   j++;
               } else {
                   i--;
               }
           }
           
           return false;
       }
   }