劍指 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。

限制：
0 <= n <= 1000

0 <= m <= 1000

1、暴力法

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        for (int i = 0 ; i < matrix.length ; i++){
            for (int j = 0 ; j < matrix[i].length ; j++){
                if(matrix[i][j] == target){
                    return true;
 

                }
            }
        }
        return false;
    }
}

2、線性查詢

該思路由Leecode官網提供

具體思路：從二維陣列的右上角開始查詢。如果當前元素等於目標值，則返回 true。如果當前元素大於目標值，則移到左邊一列。如果當前元素小於目標值，則移到下邊一行。

• 若陣列為空，返回 false
• 初始化行下標為 0，列下標為二維陣列的列數減 1
• 重複下列步驟，直到行下標或列下標超出邊界
  ○獲得當前下標位置的元素 num
  ○如果num和target相等，返回 true
  ○如果 num大於target，列下標減 1
  
  ○如果 num小於 target，行下標加 1
• 迴圈體執行完畢仍未找到元素等於 target，說明不存在這樣的元素，返回 false

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