题目名称
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一下):