自我修炼_中级算法篇_leetcode_第2题

矩阵置零
给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]
示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
 

提示：

m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1
 

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-medium/xvmy42/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

这道题首先的想法是，遍历这个二维vector容器，得到其0位置的索引，然后将对应位置都记为0.

 int n = matrix[0].size(); //列
        int m = matrix.size();  //行
        vector<int> row(m),col(n);
        for(int i = 0;i<n;i++){
            for(int j = 0;j<m;j++){
                if(matrix[i][j] == 0){
                    row[i]=1;
                    col[j]=1;
                }
            }
        }

在这个代码里，我们声明了两个一维容器，vector,分别作为记录matrix的行和列，当matrix中的元素为0时，我们给row和col分别赋值为1.为什么这样赋值呢，因为这道题是让你将原matrix的所有0的行和列值都赋值为0.

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

 我们以例一为例，也就是row[1] = 1 ,col[1] = 1;matrix[1][1]=0;

我们需要的是将matrix[0][1]=0,matrix[1][1]=0,matrix[2][1] = 0;matrix[1][0] = 0,matrix[1][1] = 0 matrix[1][2] = 0;