原题如下:
矩阵归零
给定一个m×n矩阵,如果一个元素是0,则将其所在行和列全部元素变成0。
需要在原矩阵上完成操作。
给出一个矩阵
[
[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;
}
}
}
};