程序员面试金典 1.7 清除行列

题目

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

给定一个MxN的int[][]矩阵(C++中为vector>)mat和矩阵的阶数n，请返回完成操作后的int[][]矩阵(C++中为vector>)，保证n小于等于300，矩阵中的元素为int范围内。

我的题解

vector<vector<int> > clearZero(vector<vector<int> > mat, int n) {
// 标记需要清楚的行与列，然后执行
vector<int> vec(n, 0);
int i, j;
for(i = 0; i < n; ++ i) {
for(j = 0; j < n; ++ j) {
if(mat[i][j] == 0) {
vec[i] |= 2;
vec[j] |= 1;
}
}
}

for(i = 0; i < n; ++ i) {
if((vec[i] & 2) > 0) {
for(j = 0; j < n; j ++)
mat[i][j] = 0;
}
if((vec[i] & 1) > 0) {
for(j = 0; j < n; j ++)
mat[j][i] = 0;
}
}

return mat;
}