矩阵归零-LintCode

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。
需要在原矩阵上完成操作。
样例：
给出一个矩阵
[
[1, 2],
[0, 3]
]
返回
[
[0, 2],
[0, 0]
]
挑战 ：
你是否使用了额外的空间？
一个直接的解决方案是使用 O(MN) 的额外空间，但这并不是一个好的方案。
一个简单的改进方案是使用 O(M + N) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗？

#ifndef C162_H#define C162_H#include<iostream>#include<vector>using namespace std;class Solution {public:    /*    * @param matrix: An integer    * @return:    */    void setZeroes(vector<vector<int>>& matrix) {        // write your code here        int len = matrix.size();        if (len <= 0)            return;        int size = matrix[0].size();        if (size <= 0)            return;        bool rowzero = false, colzero = false;        for (int i = 0; i < len; ++i)        {            if (matrix[i][0] == 0)            {                colzero = true;                break;            }        }        for (int j = 0; j < size; ++j)        {            if (matrix[0][j] == 0)            {                rowzero = true;                break;            }        }        for (int i = 1; i < len; ++i)        {            for (int j = 1; j < size; ++j)            {                if (matrix[i][j] == 0)                {                    matrix[i][0] = 0;                    matrix[0][j] = 0;                }            }        }        for (int j = 1; j < size; ++j)        {            if (matrix[0][j] == 0)            {                for (int i = 0; i < len; ++i)                {                    matrix[i][j] = 0;                }            }        }        for (int i = 1; i < len; ++i)        {            if (matrix[i][0] == 0)            {                for (int j = 0; j < size; ++j)                {                    matrix[i][j] = 0;                }            }        }        if (rowzero)        {            for (int j = 0; j < size; ++j)                matrix[0][j] = 0;        }        if (colzero)        {            for (int i = 0; i < len; ++i)                matrix[i][0] = 0;        }    }};#endif