【LeetCode】 73. 矩阵置零 空间O(1)

题目

题目传送门：传送门（点击此处）

题解

1. 空间复杂度为O(mn)的解决方案很容易思考出来
2. 空间复杂度为O(m+n)的解决方案也比较容易，创建两个数组长度分别为m和n，用来记录行或者列中出现过的0
3. 空间复杂度为O(1)的解决方案就是在 2 的基础上，把这个数组放在了矩阵中来记录

我的思路

1. 遍历找到第一个 0，并且把这一行和这一列作为记录有 0 的行和列
   
   找到的第一个 0 在第一行，第二列，所以第一行和第二列，就作为记录的行和列
   
2. 继续向下找，每找到 0 ，就在这一行中进行记录，
   
3. 记录好了之后，先置零列
   
4. 再置零行
   

代码

class Solution {
    public void setZeroes(int[][] matrix) {
        int rows = matrix.length;
        int cols = matrix[0].length;
        int recRow = -1;
        int recCol = -1;
        // 把有 0 的行和列都记录到 recRow 和 recCol 中
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (matrix[i][j] == 0) {
                    if (recRow == -1 && recCol == -1) {
                        recRow = i;
                        recCol = j;
                    } else {
                        matrix[recRow][j] = 0;
                        matrix[i][recCol] = 0;
                    }
                }
            }
        }
        // 如果矩阵中没有 0 ，就直接返回
        if (recCol == -1 && recRow == -1) return;
        // 将 recRow 记录的所有的行设置为 0
        for (int j = 0; j < cols; j++) {
            if (j != recCol) {
                if (matrix[recRow][j] == 0) {
                    for (int i = 0; i < rows; i++) {
                        matrix[i][j] = 0;
                    }
                } else {
                    matrix[recRow][j] = 0;
                }
            }
        }
        // 将 recCol 记录的所有的列设置为 0
        for (int i = 0; i < rows; i++) {
            if (i != recRow) {
                if (matrix[i][recCol] == 0) {
                    for (int j = 0; j < cols; j++) {
                        matrix[i][j] = 0;
                    }
                } else {
                    matrix[i][recCol] = 0;
                }
            }
        }

    }
}