careercup-数组和字符串1.7

1.7 编写一个算法，若M*N矩阵中某个元素为0，则将其所在的行与列清零。

类似于leetcode中的 Set Matrix Zeroes

C++实现代码：

#include<iostream>
#include<vector>
using namespace std;

void setMatricZero(vector<vector<int> > &matrix)
{
    if(matrix.empty()||matrix[0].empty())
        return;
    int i,j;
    int m=matrix.size();
    int n=matrix[0].size();
    int rflag=0;
    int cflag=0;
    for(i=0; i<m; i++)
        if(matrix[i][0]==0)
            rflag=1;
    for(j=0; j<n; j++)
        if(matrix[0][j]==0)
            cflag=1;
    for(i=1; i<m; i++)
        for(j=1; j<n; j++)
            if(matrix[i][j]==0)
            {
                matrix[i][0]=0;
                matrix[0][j]=0;
            }
    for(i=1; i<m; i++)
        for(j=1; j<n; j++)
        {
            if(matrix[i][0]==0||matrix[0][j]==0)
                matrix[i][j]=0;
        }
    if(rflag)
    {
        for(i=0; i<m; i++)
            matrix[i][0]=0;
    }
    if(cflag)
    {
        for(j=0; j<n; j++)
            matrix[0][j]=0;
    }
}

int main()
{
    vector<vector<int> > matrix=
    {
        {0,12,13,14},
        {15,16,17,18},
        {19,0,21,22},
        {23,24,25,26}
    };
    setMatricZero(matrix);
    for(auto a:matrix)
    {
        for(auto t:a)
            cout<<t<<" ";
        cout<<endl;
    }
}