[LeetCode]73. 矩阵置零(java实现)模拟题
1. 题目
2. 读题(需要重点注意的东西)
思路:
要求使用原地算法,即不开辟新的空间;
我们原本需要两个数组来记录在行方向上哪些行需要刷成0,在列方向上哪些列需要刷成0;由于不能开辟额外的空间
,我们可以借用原本的矩阵来存储这些信息,即用第0列和第0行来存储哪些行与哪些列需要刷成0,将要刷成0的行和列的值置为0。
这样做,如果直接行和列都从0遍历到结尾的话会造成本来不该置0的行或列置为0,如图:
因此我们按如下方式遍历:
-
记录原本第0行和第0列是否有0
-
将需要置0的行标记在
行首matrix[i][0]
,将需要置0的列标记在列首matrix[0][j]
,- 再遍历行:从第
1
行遍历到第最后一行,如果该行的行首matrix[i][0]
为0,将本行置0; - 再遍历列:从第
1
列遍历到最后一列,如果该列的列首matrix[0][j]
为0,将本列置0;
- 再遍历行:从第
-
最后,如果第0行原本有0,将第0行全部刷为0;如果第0列有0,将第0列刷为0;(即若第 i 行要全部置为0,则
matrix[i][0] = 0
;若第 j 行要全部置为0,则matrix[0][j] = 0
)
3. 解法
---------------------------------------------------解法---------------------------------------------------:
class Solution {
public void setZeroes(int[][] matrix) {
int row = 1;
int col = 1;
// 记录原本第0行和第0列是否有0
for(int i = 0;i < matrix[0].length;i++)
if(matrix[0][i] == 0){
row = 0;
break;
}
for(int i = 0;i < matrix.length;i++){
if(matrix[i][0] == 0){
col = 0;
break;
}
}
// 标记哪些行和列要刷成0
for(int i = 0;i < matrix.length;i++){
for(int j = 0;j < matrix[0].length;j++){
if(matrix[i][j] == 0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
// 将标记好的行和列刷为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;
}
}
// 看列
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;
}
}
// 最后刷第0行和第0列
if(row == 0)
for(int i = 0;i < matrix[0].length;i++)
matrix[0][i] = 0;
if(col == 0)
for(int i = 0;i < matrix.length;i++)
matrix[i][0] = 0;
}
}
可能存在的问题:
4. 可能有帮助的前置习题
5. 所用到的数据结构与算法思想
6. 总结
模拟题,但是用矩阵本身来存储信息的思路很巧妙。