题目

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?

分析

给定一个数组，如果某个元素为0，将其整行与整列都改为0.题目提示可以通过常数内存空间就可解决。我是用m+n的数组来表示某行或某列是否需要改为0.
有些人用matrix的第一行与第一列保存该行或该列是否存在0，这样就不需要额外的内存空间了。

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;
            }
        }
    }
}