LC 零矩阵

最新推荐文章于 2024-05-21 20:29:42 发布

学庭

最新推荐文章于 2024-05-21 20:29:42 发布

阅读量540

点赞数 9

分类专栏： Leetcode 文章标签：矩阵算法线性代数 c++

本文链接：https://blog.csdn.net/Kruskual/article/details/135610504

版权

Leetcode 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

LC 零矩阵

题目描述：

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

题目实例：

示例一：
输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例二：
输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

审题：

本题对空间复杂度无要求，我们可以申请额外的空间来解题。

我的大致思路：

第一步：申请同样大小的空间V，用以记录有0的点

matrix = 
[
  [1,2,3],
  [4,0,6],
  [7,8,9]
]

===》
V = 
[
  [0,0,0],
  [0,1,0],
  [0,0,0]
]

第二步：遍历V，找到有0的点，对原数组的行和列赋值为0。

matrix = 
[
  [1,4,7],
  [2,0,8],
  [3,6,9]
]

===》
matrix = 
[
  [7,0,1],
  [0,0,0],
  [9,0,3]
]

代码如下：

void setZeroes(vector<vector<int>>& matrix) {
    int m = matrix.size();
    int n = matrix[0].size();
    vector<vector<int>> myvector;
    myvector.resize(m, vector<int>(n, 0));

    for(int i = 0; i < m; ++i){
        for(int j = 0; j < n; ++j){
            if(matrix[i][j] == 0){
                myvector[i][j] = 1;
            }else{
                myvector[i][j] = 0;
            }
        }
    }

    for(int i = 0; i < m; ++i){
        for(int j = 0; j < n; ++j){
            if(myvector[i][j] == 1){
               for(int x = 0; x < n; ++x){
                matrix[i][x] = 0;
               }
               for(int x = 0; x < m; ++x){
                matrix[x][j] = 0;
               }               
            }
        }
    }    
}