lintcode刷题——矩阵归零

原题如下：

矩阵归零 

•  描述
•  笔记
•  数据
•  评测

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。

需要在原矩阵上完成操作。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出一个矩阵

[
  [1, 2],
  [0, 3]
]

返回

[
  [0, 2],
  [0, 0]
]

做题思路：

1、题目比较简单，只要有0那么该行该列的所有数字都设为0，所以只要找到为0的数的位置即可，注意会有多个0存在的情况；

2、定义两个vector存放为0的数的横纵坐标，然后将对应的坐标的行列上的所有数设为0即可。

具体的C++代码如下：

class Solution {
public:
    /*
     * @param matrix: A lsit of lists of integers
     * @return: 
     */
    void setZeroes(vector<vector<int>> &matrix) {
        // write your code here
     int len = matrix.size();
if (len == 0)
return;
int longs = matrix[0].size();
if (longs == 0)
return;
int i, j;
vector<int> px, py;
for (i = 0; i < len; i++)
{
for (j = 0; j < longs; j++)
{
if (matrix[i][j] == 0)
{
px.push_back(i);
py.push_back(j);
}
}
}
int lx=px.size(),ly=py.size();
for (i = 0; i < lx; i++)
{
for (j = 0; j < longs; j++)
{
matrix[px[i]][j] = 0;
}
}
for (i = 0; i < len; i++)
{
for (j = 0; j < ly; j++)
{
matrix[i][py[j]] = 0;
}
}
    }
};