[剑之offer] 04 二维数组中的查找

一、问题？

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

2、示例

a: 现有矩阵 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]

]

b: 给定 target = 5，返回 true。

c: 给定 target = 20，返回 false。

3、限制：

0 <= n <= 1000

0 <= m <= 1000

二、解题

1、思路 暴力求解 循环遍历

2、利用数组的有序规律

如果当前元素大于目标值，说明当前元素的下边的所有元素都一定大于目标值，因此往下查找不可能找到目标值，往左查找可能找到目标值。如果当前元素小于目标值，说明当前元素的左边的所有元素都一定小于目标值，因此往左查找不可能找到目标值，往下查找可能找到目标值。

排除数组中的部分元素。

package com.haoxiansheng.demo01.SwordfingerOffer;

import lombok.extern.slf4j.Slf4j;

/**
 * @author flame
 * @data 2020/10/14
 */
@Slf4j
public class FindNumberIn2DArray {
    public static void main(String[] args) {
        int[][] test = {
                {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}};
        log.info("findNumberIn2DArray=>{}", findNumberIn2DArray(test, 19));
        log.info("findNumberIn2DArray2=>{}", findNumberIn2DArray2(test, 19));
    }

    // 根据数组的排序规律求解
    //时间复杂度：O(n+m)。访问到的下标的行最多增加 n 次，列最多减少 m 次，因此循环体最多执行 n + m 次。
    //空间复杂度：O(1)。
    public static 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];
            log.info("matrix[row][column]=>{},{}", row, column);
            if (num == target) {
                return true;
            } else if (num > target) {
                column--;
            } else {
                row++;
            }
        }
        return false;
    }

    //  暴力求解
    //时间复杂度：O(nm)O(nm)。二维数组中的每个元素都被遍历，因此时间复杂度为二维数组的大小。
    // 空间复杂度：O(1)O(1)。
    public static boolean findNumberIn2DArray2(int[][] matrix, int target) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            return false;
        }
        int rows = matrix.length, columns = matrix[0].length;
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < columns; j++) {
                log.info("(matrix[i][j]=>{}{}", i, j);
                if (matrix[i][j] == target) {
                    return true;
                }
            }
        }
        return false;
    }

}