9,638
社区成员
发帖
与我相关
我的任务
分享
编写一种算法,若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]
]
row col分别记录是否有第一行为0的位置,因为其他节点为0的话,把其他节点对应的行和列的第一个位置置为0.
class Solution {
public void setZeroes(int[][] matrix) {
boolean row=false,col=false;
int m=matrix.length,n=matrix[0].length;
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[i][j]==0){
matrix[i][0]=0;
matrix[0][j]=0;
if(i==0) row=true;
if(j==0) col=true;
}
}
}
for(int i=1;i<m;i++)
if(matrix[i][0]==0)
for(int j=0;j<n;j++) matrix[i][j]=0;
for(int i=1;i<n;i++)
if(matrix[0][i]==0)
for(int j=0;j<m;j++) matrix[j][i]=0;
if(row)
for(int i=0;i<n;i++) matrix[0][i]=0;
if(col)
for(int i=0;i<m;i++) matrix[i][0]=0;
// System.out.println("[");
// for(int i=0;i<m;i++){
// System.out.print("[");
// for(int j=0;j<n;j++){
// if(j!=n-1) System.out.print(matrix[i][j]+",");
// else System.out.print(matrix[i][j]);
// }
// if(i!=m-1) System.out.println("],");
// else System.out.println("]");
// }
// System.out.println("]");
}
}