CCI_Q1.7

本文参考该作者文章当作编程笔记：

作者：Hawstein
出处：http://hawstein.com/posts/ctci-solutions-contents.html

一.

Q:写一个函数处理一个N×M的矩阵，如果矩阵中某个元素为0，那么把它所在的行和列都置为0.

思路：额外申请两个数组row[N],col[M]，如果元素s[i][j]等于0，那么row[i]和col[j]都等于0，最后重新遍历数组，将行列对应的元素设为0.

CODE：

#include<stdio.h>
#include<memory.h>
#define N 4
#define M 4
void zero(int s[][M] )
{
    int row[N],col[M],i,j;
    memset(row,0,sizeof(row));
    memset(col,0,sizeof(col));
    for(i=0;i<N;i++)
        for(j=0;j<M;j++)
        {
            if(s[i][j]==0)
            {
                row[i]=1;
                col[j]=1;
            }
        }
    for(i=0;i<N;i++)
        for(j=0;j<M;j++)
        {
            if(row[i]==1||col[j]==1)
                s[i][j]=0;
        }

}
int main(void)
{
    int s[N][M]={{1,2,3,0},{6,2,3,6},{9,8,8,4},{1,0,2,3}};
    int i,j;
    for(i=0;i<N;i++)
    {
        for(j=0;j<M;j++)
            printf("%4d",s[i][j]);
        printf("\n");
    }
    printf("\n");
    zero(s);
    for(i=0;i<N;i++)
    {
        for(j=0;j<M;j++)
            printf("%4d",s[i][j]);
        printf("\n");
    }

    return 0;
}