一. 这题学习题,借鉴官方题解.
作者:LeetCode
链接:https://leetcode-cn.com/problems/set-matrix-zeroes/solution/ju-zhen-zhi-ling-by-leetcode/
二. 这题空间复杂度要求是 O(1).用三种方法.
三. 方法 1:额外存储空间方法(暴力大法要快速实现coding,一定记住这个)
想法: 如果矩阵中任意一个格子有零我们就记录下它的行号和列号,这些行和列的所有格子在下一轮中全部赋为零。
算法: 1. 我们扫描一遍原始矩阵,找到所有为零的元素。如果我们找到 [i, j] 的元素值为零,我们需要记录下行号 i 和列号 j。
2. 用两个哈希表 ,一个记录行信息一个记录列信息。3. 最后,我们迭代原始矩阵,对于每个格子检查行 r
和列 c
是否被标记过,如果是就将矩阵格子的值设为 0。
#include <iostream>
#include <vector>
#include <map>
using namespace std;
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int r = matrix.size(), c = matrix[0].size();//假设数组非空.
map<int, int> row; map<int, int> col;
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
if (matrix[i][j] == 0) {//如果是0,插入行列号
row.insert(map<int, int>::value_type(i, 1));//我喜欢的哈希表插入方法
col.insert(map<int, int>::value_type(j, 1));
}
}
}
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
//查找哈希表有无这个键可以用count来计数,不过推荐使用find(),仅一次查找,且时间复杂度为logn
if (row.count(i) > 0 || col.count(j) > 0) {
matrix[i][j] = 0;
}
}
}
}
};
复杂度分析
- 时间复杂度:O(M×N),其中 M 和 N 分别对应行数和列数。
- 空间复杂度:O(M+N)。
四. 方法 2:O(1)空间的暴力
想法:在上面的方法中我们利用额外空间去记录需要置零的行号和列号,通过修改原始矩阵可以避免额外空间的消耗。
算法:1. 遍历原始矩阵,如果发现如果某个元素 cell[i][j] 为 0,我们将第 i 行和第 j 列的所有非零元素设成很大的负虚拟值(比如说 -1000000)。注意,正确的虚拟值取值依赖于问题的约束,任何允许值范围外的数字都可以作为虚拟值。2. 最后,我们便利整个矩阵将所有等于虚拟值的元素设为 0。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int r = matrix.size(), c = matrix[0].size();//假设数组非空.
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
if (matrix[i][j] == 0) {//如果是0,相对应行列赋予一个最小值以区别其他所有值.
for (int k = 0; k < r; k++){
if(matrix[k][j]!=0)
matrix[k][j] = INT_MAX;
}
for (int k = 0; k < c; k++){
if(matrix[i][k]!=0)
matrix[i][k] = INT_MAX;
}
}
}
}
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
//将所有设置为区别的值改为0.
if (matrix[i][j] == INT_MAX) {
matrix[i][j] = 0;
}
}
}
}
};
但是由于区别值不能再原来数组出现,所以报错.
复杂度分析
时间复杂度:O((M×N)×(M+N)),其中 M 和 N 分别对应行数和列数。尽管这个方法避免了使用额外空间,但是效率很低,因为最坏情况下每个元素都为零我们需要访问所有的行和列,因此所有 (M×N) 个格子都需要访问 (M+N) 个格子并置零。
空间复杂度:O(1)
五. 第二种方法不高效的地方在于我们会重复对同一行或者一列赋零。我们可以推迟对行和列赋零的操作。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int r = matrix.size(), c = matrix[0].size();//假设数组非空.
//假设matrix[0][0]代表行的标志, 第0行和第0列最后考虑.
//只需要在相应行列的第一个位置设置标志0就好了.
//也就是相应的第0行和第0列就是标志位.
bool iscol = false;
for (int i = 0; i < r; i++) {
if (matrix[i][0] == 0)
iscol = true;
for (int j = 1; j < c; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
//先把除了第0行第0列的先处理,防止覆盖,因为第0行第0列标志位不能动.
for (int i = 1; i < r; i++) {
for (int j = 1; j < c; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0)
matrix[i][j] = 0;
}
}
//先看第0行是否需要全部置零
if (matrix[0][0] == 0) {
for (int j = 0; j < c; j++) {
matrix[0][j] = 0;
}
}
//再看第0列是否需要全部置零
if (iscol) {
for (int i = 0; i < r; i++) {
matrix[i][0] = 0;
}
}
}
};
复杂度分析
- 时间复杂度:O(M×N)
- 空间复杂度:O(1)