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

思路：

• 首先選取數組中右上角的數字，如果該數字爲要查找的數字，則查詢過程結束；
• 如果該數字大於target，則剔除該數字所在列，
• 如果該數字小於target，那就剔除該數字所在的行。
• 也就是說，如果target不在數組的右上角，那麼每一次都在數組的查找範圍中剔除一行或一列，縮小查找範圍，直到查找出target或者查詢範圍爲空。

const findNumberIn2DArray = (matrix: number[][], target: number): boolean  => {
    if (!matrix.length || matrix == null) return false;
    
    let res: boolean = false;
    let row: number = matrix.length;
    let col: number = matrix[0].length;

    if (matrix !== null && row > 0 && col > 0) {
        let rowTemp: number = 0;
        let colTemp: number = col - 1;

        while (rowTemp < row && colTemp >= 0) {
            if (matrix[rowTemp][colTemp] === target) {
                res = true;
                break;
            } else if (matrix[rowTemp][colTemp] > target) {
                --colTemp;
            } else {
                ++rowTemp;
            }
        }
    }

    return res;
};