程序员面试金典 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;
}