[LeetCode] Set Matrix Zeroes

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Follow up:

Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

分别记录两个向量x, y,保存行和列是否有0,再次遍历数组时查询对应的行和列然后修改值。
 1 class Solution {
 2 private:
 3     bool x[1000];
 4     bool y[1000];
 5 public:
 6     void setZeroes(vector<vector<int> > &matrix) {
 7         // Start typing your C/C++ solution below
 8         // DO NOT write int main() function
 9         memset(x, false, sizeof(x));
10         memset(y, false, sizeof(y));
11         
12         for(int i = 0; i < matrix.size(); i++)
13             for(int j = 0; j < matrix[i].size(); j++)
14                 if (matrix[i][j] == 0)
15                 {
16                     x[i] = true;
17                     y[j] = true;
18                 }
19                 
20         for(int i = 0; i < matrix.size(); i++)
21             for(int j = 0; j < matrix[i].size(); j++)
22                 if (x[i] || y[j])
23                     matrix[i][j] = 0;
24     }
25 };

 

posted @ 2012-11-18 10:54  chkkch  阅读(829)  评论(0编辑  收藏  举报