在有序的二维数组中查找 java

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

暴力法

直接遍历，两层循环，找到后直接返回

public boolean findOn2(int[][] matrix, int number) {
    boolean found = false;
    if (matrix != null && matrix.length > 0)
        for (int i = 0; i < matrix.length; i++)
            for (int j = 0; j < matrix[i].length; j++)
                if (matrix[i][j] == number) {
                    found = true;
                    break;
                }
    return found;
}

时间复杂度： 根据二维数组的大小
空间复杂度： O(1)

暴力法忽视了有序二维数组的条件。我们查找规律 ，发现可以首先选取数组中右上角的数字。如果该数据等于要查找的数字，则查找过程结束；如果该数字大于要查找的数字，则剔除这个数字所在的列：如果该数字小于要查找的数字，则剔除这个数字所在的行。也就是说，如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中提出一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

public boolean findNumberIn2DArray(int[][] matrix, int target) {
    boolean found = false;
    if (matrix != null && matrix.length > 0) {
        int row = 0;
        // 获取第一行的个数即是二维数组的列数
        int column = matrix[row].length - 1;
        while (row < matrix.length && column >= 0) {
            // 选取数组右上角的数字，如果等于要查找的number,查找过程结束
            if (matrix[row][column] == target) {
                found = true;
                break;
            } else
                // 如果该数字大于要找的数字，剔除该数字所在的列
                if (matrix[row][column] > target)
                    column--;
                    // 如果该数字小于要查找的数字，剔除所在行
                else if (matrix[row][column] < target)
                    row++;
        }
    }
    return found;
}

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

同样的我们也可以通过查找右下角的数字的方式。

测试用例：

@Test
public void FindTest1() {
    int[][] arr = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
    int[][] arr1 = {};
    int[][] arr2 = {{}};

    Find2DArray find2DArray = new Find2DArray();
    Assert.assertTrue(find2DArray.findOn2(arr, 7));
    Assert.assertFalse(find2DArray.findOn2(arr1, 0));
    Assert.assertFalse(find2DArray.findOn2(arr2, 1));
}

@Test
public void FindTest2() {
    int[][] arr = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
    int[][] arr1 = {};
    int[][] arr2 = {{}};

    Find2DArray find2DArray = new Find2DArray();
    Assert.assertTrue(find2DArray.findNumberIn2DArray(arr, 7));
    Assert.assertFalse(find2DArray.findNumberIn2DArray(arr1, 0));
    Assert.assertFalse(find2DArray.findNumberIn2DArray(arr2, 1));
}