题目描述
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.
Example 1:
Input:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
Output:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
Example 2:
Input:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
Output:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
找出矩阵中的0,把0所在的行和列全都置为0
思路
- 直接法
遍历找到矩阵中所有的0,把0所在的行和列所有元素设置为特定值,然后再次遍历矩阵,将被设为特定值的元素设为0 - 借助集合set存放找到的0元素所在的行号和列号
然后利用集合查询元素的方法,遍历矩阵,若找到了集合中的元素,就设为0
实现
- 直接法 :时间复杂度O(mn(m+n)) 空间复杂度O(1)
public void setZeroes(int[][] matrix) {
//直接遍历找到符合条件的,再将行列置为0
int R=matrix.length; //行数
int C=matrix[0].length; //列数
//定义一个很大的负虚数(确保不和数组中已有元素重复)
int MODIFIED=-1000000;
//遍历找0
for(int i=0;i<R;i++){
for(int j=0;j<C;j++){
if(matrix[i][j]==0){
//将第i行,第j列中不为0的元素设为很大的负虚数MODIFIED
for(int k=0;k<C;k++){
if(matrix[i][k]!=0){
matrix[i][k]=MODIFIED;
}
}
for(int k=0;k<R;k++){
if(matrix[k][j]!=0){
matrix[k][j]=MODIFIED;
}
}
}
}
}
//将被设置为MODIFIED的元素设为0
for(int i=0;i<R;i++){
for(int j=0;j<C;j++){
if(matrix[i][j]==MODIFIED){
matrix[i][j]=0;
}
}
}
}
- 借助额外空间:两个set分别保存0元素所在的行号和列号 时间复杂度O(m*n) 空间复杂度O(m+n)
public void setZeroes(int[][] matrix) {
//额外空间,借助两个集合set存放0元素所在的行号和列号
int R=matrix.length; //行数
int C=matrix[0].length; //列数
Set<Integer> rows=new HashSet<>();//存放0元素行号的集合
Set<Integer> cols=new HashSet<>();//存放0元素列号的集合
for(int i=0;i<R;i++){
for(int j=0;j<C;j++){
if(matrix[i][j]==0){
rows.add(i); //把0元素所在的所有行号加到rows集合中
cols.add(j); //把0元素所在的所有列号加到cols集合中
}
}
}
for(int i=0;i<R;i++){
for(int j=0;j<C;j++){
if(rows.contains(i) || cols.contains(j)){ //若i或j是集合中的元素,就置0
matrix[i][j]=0;
}
}
}
}