題目

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(m n) 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?

分析

給定一個數(shù)組，如果某個元素為0，將其整行與整列都改為0.題目提示可以通過常數(shù)內(nèi)存空間就可解決。我是用m+n的數(shù)組來表示某行或某列是否需要改為0.
有些人用matrix的第一行與第一列保存該行或該列是否存在0，這樣就不需要額外的內(nèi)存空間了。

void setZeroes(int** matrix, int matrixRowSize, int matrixColSize) {
    int rowflag[matrixRowSize],colflag[matrixColSize];
    for(int i=0;i<matrixRowSize;i++)
        rowflag[i]=0;
    for(int j=0;j<matrixColSize;j++)
        colflag[j]=0;
    for(int i=0;i<matrixRowSize;i++)
    {
        for(int j=0;j<matrixColSize;j++)
        {
            if(matrix[i][j]==0)
            {
                rowflag[i]=1;
                colflag[j]=1;
            }
        }
    }
    for(int i=0;i<matrixRowSize;i++)
    {
        for(int j=0;j<matrixColSize;j++)
        {
            if(rowflag[i]==1||colflag[j]==1)
            {
                matrix[i][j]=0;
            }
        }
    }
}