LeetCode面试题 零矩阵

LeetCode面试题 零矩阵

题目

编写一种算法，若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]
]

分析

从题目中我们分析可以知道，对于M × N矩阵中某个元素为0,则她对应的行和列所有元素都被清零。针对这个问题我们可以先遍历一次矩阵，找到哪些元素为0并将其坐标（行列）通过一个二维数组存储起来，然后通过得到的行列坐标，将对应的行列一次进行清零。

代码实现

class Solution {
    public void setZeroes(int[][] matrix) {
    //矩阵的列宽度，矩阵行宽度
        int c=matrix.length,r=matrix[0].length;
        //定义动态数组，存储元素为零的下标
        List<int[]> arry=new ArrayList<>();
        for(int i=0;i<c;i++){
            for(int j=0;j<r;j++){
                if(matrix[i][j]==0){
                    arry.add(new int[]{i,j});
                }
            }
        }
        for(int t=0;t<arry.size();t++){
            for(int l=0;l<r;l++){
            //行清零
                matrix[arry.get(t)[0]][l]=0;
                }
                for(int l2=0;l2<c;l2++){
               //列清零
                matrix[l2][arry.get(t)[1]]=0;
            } 
        }
    }
}

结果

干哈呢？不打算点个赞在走！