package arrayandstring;
import java.util.Arrays;
import java.util.HashSet;
//一个直接的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
//一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
//你能想出一个常数空间的解决方案吗?
public class SetMatrixZeroes {
public static void main(String[] args) {
int[][] matrix = {
{0,1,2,0},
{3,4,5,2},
{1,3,1,5}
};
SetMatrixZeroes test = new SetMatrixZeroes();
test.setZeroes(matrix);
for (int[] link : matrix) {
System.out.println(Arrays.toString(link));
}
}
public void setZeroes(int[][] matrix) {
int n = matrix.length;
int m = matrix[0].length;
//O(m + n) 的额外空间
HashSet<Integer> row = new HashSet<>(n);
HashSet<Integer> col = new HashSet<>(m);
for(int i = 0; i < n; i ++) {
for (int j = 0; j < m; j ++) {
if (matrix[i][j] == 0) {
row.add(i);
col.add(j);
}
}
}
//设置行
/*Iterator it = row.iterator();
while (it.hasNext()) {
Object r = it.next();
int
}*/
for (int r : row) {
for (int i = 0; i < m; i ++) {
matrix[r][i] = 0;
}
}
for (int c : col) {
for (int i = 0; i < n; i ++) {
matrix[i][c] = 0;
}
}
}
public void setZeroes02(int[][] matrix) {
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 < R; j ++) {
if (matrix[i][j] == 0) {
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;
}
}
}
}
}
for (int i = 0; i < R; i ++) {
for (int j = 0; j < R; j++) {
if (matrix[i][j] == MODIFIED) {
matrix[i][j] = 0;
}
}
}
}
}
矩阵置零的两种思路及JAVA实现
最新推荐文章于 2024-06-28 18:34:31 发布