162 · 矩阵归零

162 · 矩阵归零

描述

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。

需要在原矩阵上完成操作。
样例

样例 1:

输入:[[1,2],[0,3]]
输出:[[0,2],[0,0]]

样例 2:

输入:[[1,2,3],[4,0,6],[7,8,9]]
输出:[[1,0,3],[0,0,0],[7,0,9]]

挑战

你是否使用了额外的空间？
一个直接的解决方案是使用 O(MN) 的额外空间，但这并不是一个好的方案。
一个简单的改进方案是使用 O(M + N) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗？

void setZeroes(vector<vector<int>> &matrix) {

        // write your code here

        int rowSize = matrix.size();

        if(rowSize <=0)

            return ;

        int colSize = matrix[0].size();

        vector<bool> rowVec(rowSize,false);

        vector<bool> colVec(colSize,false);

 

        for(int i =0;i<rowSize;i++)

        {

            for(int j=0;j<colSize;j++)

            {

                if(matrix[i][j] == 0 )

                {

                    rowVec[i] = true;

                    colVec[j] = true;

                }

            }

        }

 

         for(int i =0;i<rowSize;i++)

        {

            for(int j=0;j<colSize;j++)

            {

                if(true == rowVec[i] || true == colVec[j])

                {

                  matrix[i][j] = 0;

                }

            }

        }

    }