18.矩阵置零

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

示例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
• -2³¹ <= matrix[i][j] <= 2³¹ - 1

代码：

class Solution {
    public void setZeroes(int[][] matrix) {
        //m为行数，n为列数
        int m = matrix.length,n = matrix[0].length;
        //flag用于标记第一列是否全为0
        int flag = 0;
        //第一列只要存在0，标记就记为1
        for(int i=0;i<m;i++){
            if(matrix[i][0]==0){
                flag=1;
                break;
            }
        }
        //matrix[0][0]作为第一行的标记
        //第一行只要存在0，matrix[0][0]就记为0
        for(int i=0;i<n;i++){
            if(matrix[0][i]==0){
                matrix[0][0]=0;
                break;
            }
        }
        //处理除了第一行和第一列之外的元素
        //matrix[i][j]为0，就将行首和行尾元素置为0
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        //处理除了第一行和第一列之外的元素
        //只要行首和行尾元素置为0，matrix[i][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;
            }
        }
        //matrix[0][0]为0，则第一行全部置为0
        if(matrix[0][0]==0){
            for(int i=1;i<n;i++)matrix[0][i]=0;
        }
        //flag为1，则第一列全部置为0
        if(flag==1){
            for(int i=0;i<m;i++)matrix[i][0]=0;
        }
    }
}