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;
}
}
    }
};