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