1. 题目
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2:
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zero-matrix-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.题解
- 一个直接的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
- 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
- 常数空间才最香O(1)。
思路
- 用第一行的元素表示每一列是否存在0,第一列的元素表示每一行是否存在0
- 同时要处理第一行第一列本身存在0的情况
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
if(matrix.size()==0 || matrix[0].size() == 0)
return;
bool firstRowIsZero = false,firstColIsZero = false;
for(int i =0;i<matrix.size();i++){
for(int j=0;j<matrix[0].size();j++){
if(i!=0 && j!=0 && matrix[i][j]==0){
matrix[i][0]=0;
matrix[0][j]=0;
}else if(matrix[i][j]==0){
firstRowIsZero = i==0? true:firstRowIsZero;
firstColIsZero = j==0? true:firstColIsZero;
}
}
}
for(int i=1;i<matrix.size();i++){
for(int j=1;j<matrix[0].size();j++){
if(matrix[0][j]==0 || matrix[i][0] == 0){
matrix[i][j]=0;
}
}
}
//第一列含0
if(firstColIsZero){
for(int i=0;i<matrix.size();i++){
matrix[i][0] = 0;
}
}
//第一行含0
if(firstRowIsZero){
for(int j=0;j<matrix[0].size();j++){
matrix[0][j] = 0;
}
}
}
};
参考:https://blog.csdn.net/qq_21201267/article/details/104518002
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
if(matrix.size()==0 || matrix[0].size() == 0)
return;
//先遍历除第一行,第一列外的区域
int i, j, m = matrix.size(), n = matrix[0].size();
bool r0c0Zero = (matrix[0][0] == 0);
bool r0 = 0, c0 = 0;
if(!r0c0Zero)
{
for(i = 1; i < m; i++)
if(matrix[i][0] == 0)
{
c0 = true;//第一列需要后序变0
break;
}
for(j = 1; j < n; j++)
if(matrix[0][j] == 0)
{
r0 = true;//第一行需要后序变0
break;
}
}
for(i = 1; i < m; i++)
for(j = 1; j < n; j++)
{
if(matrix[i][j] == 0)
{
matrix[0][j] = 0;//用第一行标记是否删除列
matrix[i][0] = 0;//用第一列标记是否删除行
}
}
for(i = 1; i < m; i++)
{
if(matrix[i][0] == 0)//删除行
for(j = 1; j < n; j++)
matrix[i][j] = 0;
}
for(j = 1; j < n; j++)
{
if(matrix[0][j] == 0)//删除列
for(i = 1; i < m; i++)
matrix[i][j] = 0;
}
if(r0c0Zero)//删除第一行,第一列
{
for(i = 1; i < n; i++)
matrix[0][i] = 0;
for(i = 1; i < m; i++)
matrix[i][0] = 0;
}
else
{
if(r0)//删除第一行
{
for(j = 0; j < n; j++)
matrix[0][j] = 0;
}
if(c0)//删除第一列
{
for(i = 0; i < m; i++)
matrix[i][0] = 0;
}
}
}
};
下面这个比上面的快了将近一倍!