面试题4：二维数组中的查找

题目、二维数组中的查找

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

方法1：暴力遍历
若不考虑矩阵是否有序，直接遍历二维数组，暴力求解

• 时间复杂度O(nm)
• 空间复杂度O(1)

public boolean findNumberIn2DArray(int[][] matrix, int target) {
    // 矩阵判定
    if (matrix.length <= 0 || matrix[0].length <= 0) {
        return false;
    }
    // 定义行列和判断标志
    int row = matrix.length;
    int col = matrix[0].length;
    boolean flag = false;
    for (int i = 0; i < row; i++) {
        if (flag) {
            break;
        }
        for (int j = 0; j < col; j++) {
            if (matrix[i][j] == target) {
                flag = true;
                break;
            }
        }
    }
    return flag;
}

方法2：右上角剔除法
每次都判断右上角的数字，如果右上角的数字大于要找到的数字target，则剔除最后一列；如果右上角的数字等于target，找到直接退出；如果右上角的数字小于target，则剔除这一行。不断循环，若i或j小于0，则没找到

• 时间复杂度O(n+m)
• 空间复杂度O(1)

public boolean findNumberIn2DArray(int[][] matrix, int target) {
    // 行列数大于0
    if (matrix.length == 0 || matrix[0].length == 0) {
        return false;
    }
    // 定义标志和行列数
    boolean flag = false;
    int row = matrix.length - 1;
    int col = matrix[0].length - 1;
    int rowIndex = 0;    //用作行游标
    while (rowIndex <= row && col >= 0){
        if (matrix[rowIndex][col] > target) { // 右上角数字大于target，剔除这一列，列游标--
            col--;
        } else if (matrix[rowIndex][col] < target) { // 右上角数字小于target，剔除这一行，行游标++
            rowIndex++;
        } else {
            flag = true;
            break;
        }
    }
    return flag;
}