开发者社区人工智能文章正文

careercup-数组和字符串1.7

2014-12-03 661

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 1.7 编写一个算法，若M*N矩阵中某个元素为0，则将其所在的行与列清零。类似于leetcode中的 Set Matrix Zeroes C++实现代码： #include #include using namespace std; void setMatricZero(vector &matrix) { if(matrix.

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