标签:维数 简单 其他 需要 记录 变量 row java 根据
/**
* 给定一个 m x n 的矩阵,如果一个元素为 0 ,
* 则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
*
* 一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
* 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
* 你能想出一个仅使用常量空间的解决方案吗?
*
*
*/
/**
*
* @param matrix 二维数组
*/
public void setZeroes(int[][] matrix) {
//定义两个变量判断第一行和第一列是否有0,如果有0则置为true
boolean row0 = false, col0 = false;
//判断第一行是否有0
for (int i = 0; i < matrix[0].length; i++) {
if (matrix[0][i] == 0) {
col0 = true;
break;
}
}
//判断第一列是否有0
for (int i = 0; i < matrix.length; i++) {
if (matrix[i][0] == 0) {
row0 = true;
break;
}
}
//然后从第二行第二列开始扫描,使用第一行第一列记录有0的位置
for (int i = 1; i < matrix.length; i++) {
for (int j = 1; j < matrix[i].length; j++) {
if (matrix[i][j] == 0) {
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
//扫描第一行,将相应的位置置零
for (int i = 1; i < matrix[0].length; i++) {
if (matrix[0][i] == 0) {
for (int j = 0; j < matrix.length; j++) {
matrix[j][i] = 0;
}
}
}
//扫描第一列,将相应的位置置零
for (int i = 1; i < matrix.length; i++) {
if (matrix[i][0] == 0) {
for (int j = 0; j < matrix[i].length; j++) {
matrix[i][j] = 0;
}
}
}
//判断第一行和第一列是否有0
if (col0) {
for (int i = 0; i < matrix[0].length; i++) {
matrix[0][i] = 0;
}
}
if (row0) {
for (int i = 0; i < matrix.length; i++) {
matrix[i][0] = 0;
}
}
}
标签:维数 简单 其他 需要 记录 变量 row java 根据
原文地址:https://www.cnblogs.com/mx-info/p/14865435.html