零矩阵

零矩阵

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

示例1：
输入：
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出：
[
[1,0,1],
[0,0,0],
[1,0,1]
]

示例2：
输入：
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出：
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]

思路

• 创建长度可变数组r[],c[]，分别存储矩阵元素值为零的横纵坐标；
• 遍历r[]，固定行数，将该行所有元素置0；
• 遍历c[]，固定列数，将该列所有元素置0；

代码

public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        List<Integer> r = new ArrayList<Integer>();
        List<Integer> c = new ArrayList<Integer>();
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(matrix[i][j]==0){
                    r.add(i);
                    c.add(j);
                }   
            }
        }
        for(int i=0;i<r.size();i++){
            for(int j=0;j<n;j++){
                matrix[r.get(i)][j]=0;
            }
        }
        for(int i=0;i<c.size();i++){
            for(int j=0;j<m;j++){
                matrix[j][c.get(i)]=0;
            }
        }
    }