java算法训练------ LeetCode 精选 TOP 面试题------搜索二维矩阵 II、矩阵置零

搜索二维矩阵 II

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：

   每行的元素从左到右升序排列。
    每列的元素从上到下升序排列。

示例 1：

输入：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

示例 2：

输入：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 = 20
输出：false

提示：

m == matrix.length
n == matrix[i].length
1 <= n, m <= 300
-109 <= matix[i][j] <= 109
每行的所有元素从左到右升序排列
每列的所有元素从上到下升序排列
-109 <= target <= 109

思路：他是一个有序的矩阵数据，那么我们只需要遍历第一行，然后配合列来查找即可，使用二分法让查询更快

代码：

public boolean searchMatrix(int[][] matrix, int target) {
    int m = matrix.length;
    int n = matrix[0].length;
    for (int i = 0; i < n; i++) {
        if (target == matrix[0][i] || target == matrix[0][n - 1]) {
            return true;
        }
        if (m - 1 > 0) {
            if (target == matrix[m-1][0] || target == matrix[m - 1][i])
                return true;
        }
        if (target > matrix[0][i]) {
            // 如果小于横最大值
            if (target < matrix[0][n - 1]) {
                int left = i;
                int right = n - 1;
                while (left < right) {
                    int mid = (left + right) / 2;
                    if (left == mid || right == mid) {
                        break;
                    }
                    if (target < matrix[0][mid]) {
                        right = mid;
                    } else if (target > matrix[0][mid]) {
                        left = mid;
                    } else {
                        return true;
                    }
                }
            }
            // 如果小于竖最大值
            if (m - 1 > 0 && target < matrix[m - 1][i]) {
                int left = 0;
                int right = m - 1;
                while (left < right) {
                    int mid = (left + right) / 2;
                    if (left == mid || right == mid) {
                        break;
                    }
                    if (target < matrix[mid][i]) {
                        right = mid;
                    } else if (target > matrix[mid][i]) {
                        left = mid;
                    } else {
                        return true;
                    }
                }
            }
        }
    }
    return false;
}

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-a-2d-matrix-ii

---

---

矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

进阶：

一个直观的解决方案是使用  O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗？

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

提示：

m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1

思路：遍历中间的矩阵中包含0 的x，y，并且将他们的[x][0]，[0][y]记录成0，之后通过两个for循环进行数组修改，这样的好处是为了防止重复操作，但是这里我们进行的只是内部的矩阵，所以在这边离之前，需要记录原矩阵中在[x][0]，[0][y]是否原本就存在0，存在则需要去修改数据。

代码：

public void setZeroes(int[][] matrix) {
    int m = matrix.length, n = matrix[0].length;
    boolean col = false, row = false;
    for (int i = 0; i < m; i++) {
        if (matrix[i][0] == 0) {
            col = true;
        }
    }
    for (int j = 0; j < n; j++) {
        if (matrix[0][j] == 0) {
            row = true;
        }
    }
    for (int i = 1; i < m; i++) {
        for (int j = 1; j < n; j++) {
            if (matrix[i][j] == 0) {
                matrix[i][0] = matrix[0][j] = 0;
            }
        }
    }
    for (int i = 1; i < m; i++) {
        for (int j = 1; j < n; j++) {
            if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                matrix[i][j] = 0;
            }
        }
    }
    if (col) {
        for (int i = 0; i < m; i++) {
            matrix[i][0] = 0;
        }
    }
    if (row) {
        for (int j = 0; j < n; j++) {
            matrix[0][j] = 0;
        }
    }
}

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/set-matrix-zeroes