給定一個 m x n 的矩陣,如果一個元素為 0,則將其所在行和列的所有元素都設為 0。請使用原地算法。
示例 1:
輸入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
輸出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
</pre>
示例 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]
]</pre>
進階:
- 一個直接的解決方案是使用 O(mn) 的額外空間,但這并不是一個好的解決方案。
- 一個簡單的改進方案是使用 O(m + n) 的額外空間,但這仍然不是最好的解決方案。
- 你能想出一個常數(shù)空間的解決方案嗎?
class Solution {
public void setZeroes(int[][] matrix) {
List<Integer> row=new ArrayList<>();
List<Integer> col=new ArrayList<>();
for(int i=0;i<matrix.length;++i){
for(int j=0;j<matrix[0].length;++j){
if(matrix[i][j]==0){
row.add(i);
col.add(j);
System.out.println(i+""+j);
}
}
}
// System.out.println(row.length+" "+col.length);
for(int i=0;i<row.size();++i){
for(int j=0;j<matrix[0].length;++j){
matrix[row.get(i)][j]=0;
}
for(int k=0;k<matrix.length;++k){
matrix[k][col.get(i)]=0;
}
}
}
}
class Solution {
public void setZeroes(int[][] matrix) {
int row=matrix.length;
int col=matrix[0].length;
boolean fr=false;
boolean fc=false;
for(int i=0;i<matrix.length;++i){
for(int j=0;j<matrix[0].length;++j){
if(matrix[i][j]==0){
if(i==0){
fr=true;
}
if(j==0){
fc=true;
}
matrix[0][j]=0;
matrix[i][0]=0;
}
}
}
for(int j=1;j<matrix[0].length;++j){
if(matrix[0][j]==0){
for(int i=0;i<matrix.length;++i){
matrix[i][j]=0;
}
}
}
for(int i=1;i<matrix.length;++i){
if(matrix[i][0]==0){
for(int j=0;j<matrix[0].length;++j){
matrix[i][j]=0;
}
}
}
if(fr){
for(int i=0;i<matrix[0].length;++i){
matrix[0][i]=0;
}
}
if(fc){
for(int j=0;j<matrix.length;++j){
matrix[j][0]=0;
}
}
}
}