[LeetCode] Set Matrix Zeroes (!!!！时间空间复杂度）

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.

click to show follow up.

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?

关键的问题是，当遇到A[i][j]等于0时，不能直接将第i行和第j列的元素都置为0，因为可能会影响到后面的元素，所以关键是下记录下ij的信息。

很容易想出的方法是，利用里一个行向量，一个列向量来存贮置位信息。这样空间复杂度就是O(m+n）

void SetRow(int **A, int n, int ri){
    for(int k = 0; k <n; ++k)
        A[ri][k]=0;
}

void SetCol(int **A, int m, int ci){
    for(int k = 0; k<m; ++k)
        A[k][ci]=0;
}
void setZeroes(int** A, int m, int n) {
    int i,j;
    int *c = (int *)malloc(sizeof(int)*n);
    int *r = (int *)malloc(sizeof(int)*m);
    memset(r,1,m*sizeof(int));
    memset(c,1,n*sizeof(int));
    for(i = 0; i < m; ++i)
        for(j = 0; j <n; ++j)
            if(A[i][j]==0){
                r[i]=0;
                c[j]=0;
            }
    for(i = 0; i <m; ++i)
        if(r[i]==0)
            SetRow(A,n,i);
    for(i = 0; i <n; ++i)
        if(c[i]==0)
            SetCol(A,m,i);


}

进一步进行优化，会返现上面发生干扰的问题只是干扰ij之后的元素，想不发生干扰，那么可以用第0行和第0列来代替上面的r[] c[]。第0行和第0列是否需要置零只需要两个临时变量即可。
牛逼啊~ ~

void SetRow(int **A, int n, int ri){
    for(int k = 0; k <n; ++k)
        A[ri][k]=0;
}

void SetCol(int **A, int m, int ci){
    for(int k = 0; k<m; ++k)
        A[k][ci]=0;
}
void setZeroes(int** A, int m, int n) {
    int i,j;
    int r0=1,c0=1;
    for(j=0;j<n;++j)
        if(A[0][j]==0){
            r0=0;
            break;
        }
    for(i=0;i<m;++i)
        if(A[i][0]==0){
            c0=0;
            break;
        }
    for(i = 1; i < m; ++i)
        for(j = 1; j <n; ++j)
            if(A[i][j]==0){
                A[i][0]=0;
                A[0][j]=0;
            }
    for(i = 1; i <m; ++i)
        if(A[i][0]==0)
            SetRow(A,n,i);
    for(i = 1; i <n; ++i)
        if(A[0][i]==0)
            SetCol(A,m,i);
    if(r0==0)
        SetRow(A,n,0);
    if(c0==0)
        SetCol(A,m,0);


}

空间复杂度为o（1）时间复杂度为O(2mn)