240. Search a 2D Matrix II [Medium]（剑指offer04）

从右上角或左下角开始，类似二叉搜索树BST

/**
 * 从右上角或者左上角开始运算，这样才能保证两个方向一个是变大一个是变小
 * Runtime: 4 ms, faster than 100.00%
 * Memory Usage: 44.8 MB, less than 31.44%
 */
class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int i = 0, j = matrix[0].length - 1;
        while (i < matrix.length && j >= 0) {
            if (matrix[i][j] == target) {
                return true;
            }
            if (target > matrix[i][j]) {
                i++;
            } else {
                j--;
            }
        }
        return false;
    }
}

/**
 * 2021.7.11再做，和之前代码都几乎一模一样
 * 但在剑指offer上做的多了输入矩阵为[]的edge case，所以要先判断不能直接取matrix[0].length了
 * 从右上角往左下角计算
 * 如果target大于当前，则当前整行都小于target，往下面的行找
 * 如果target小于当前，则当前整列都大于target，往左边的列找
 * 这样能持续将查找范围限定在左下区域
 */
class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if (matrix.length == 0 || matrix[0].length == 0) {
            return false;
        }
        int row = 0, col = matrix[0].length - 1;
        while (row < matrix.length && col >= 0) {
            if (matrix[row][col] == target) {
                return true;
            }
            if (matrix[row][col] > target) {
                col--;
            } else {
                row++;
            }
        }
        return false;
    }
}