Set Matrix Zeroes

题目名称
Set Matrix Zeroes—LeetCode链接

描述
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

分析
  给定一个m x n 的矩阵，如果一个元素为0，则将这个元素所在的行和列的元素全部设置为0.
  这里有一个设置元素先后的问题：如果按行遍历的过程中，搜索到了0，然后将所在的行和列的元素全部设置为0，会导致下面每一行都存在0元素，会错误的将接下来所有行的元素都设置为0。
  解决的办法是：定义一个set，将按行遍历得到的0元素所在的列的编号存放到set中，如果搜索到了0，先将其所在的行的所有元素设置为0；最后根据set中的列编号，将相应的列的所有元素全部设置为0。

C++代码

/**************************************************************

Copyright:武汉大学计算机学院B507

Author: Ryan

Date:2015-10-29

Description:Set Matrix Zeroes

**************************************************************/
#include<iostream>
#include<algorithm>
#include<set>
#include<vector>
using namespace std;
void setZeroes(vector< vector<int> >& matrix) {
    int m=matrix.size(),n=matrix[0].size();
    if(m==0 || n==0)
        return;
    set<int> col;
    for(int i=0;i<m;i++){
        //count>0表示该行出现0元素
        int count = 0;
        for(int j=0;j<n;j++){
            if(matrix[i][j]==0){
                col.insert(j);
                count++;
            }
        }
        //count>0表示该行出现0元素，将该行所有元素设置为0，用到assign方法
        if(count>0){
            matrix[i].assign(n,0);
        }
    }
    //将0元素所在的所有列输出
    for(set<int>::iterator iter=col.begin();iter!=col.end();iter++){
        cout<<*iter<<" ";
    }
    cout<<endl;
    //将0元素所在的所有列的元素设置为0
    for(set<int>::iterator iter=col.begin();iter!=col.end();iter++){
        for(int l=0;l<m;l++) {
            matrix[l][*iter]=0;
        }
    }
}

int main() {
    int a[]={1,2,3,0,4,2,3,5};
    int b[]={0,1,2,3,0,5,0,6};
    int c[]={1,2,3,4,5,6,7,8};
    vector<int> val(a,a+8);
    vector<int> val2(b,b+8);
    vector<int> val3(c,c+8);
    vector< vector<int> > res;
    res.push_back(val);
    res.push_back(val2);
    res.push_back(val3);
    setZeroes(res);
    cout<<"["<<endl;
    for(int i=0;i<res.size();i++) {
        cout<<"["<<" ";
        for(int j=0;j<res[0].size();j++) 
            cout<<res[i][j]<<" ";
        cout<<"]"<<endl;
    }
    cout<<"]"<<endl;
    return 0;
}

总结
  这里给定一个二维矩阵：

[[1,2,3,0,4,2,3,5],
[0,1,2,3,0,5,0,6],
[1,2,3,4,5,6,7,8]]

  程序运行结果和提交结果如图所示（很少有超过50%的算法，所以截个图mark一下）：