解题思路:
考虑到有些元素会被重复消除,不能实时消除,可以使用记录坐标的动态数组记录下来待消除元素,最后统一置零。消除分两步,先一行一行的消除,再一列一列的消除,相邻的三个即以上的元素都加入到数组当中即可。
参考代码:
#include <bits/stdc++.h>
using namespace std;
int n,m;
int a[35][35];
vector<pair<int, int> >v;
int main(){
cin>>n>>m;
for (int i=0; i<n; i++)
for(int j=0;j<m;j++)
{
cin>>a[i][j];
}
v.clear();
for (int i=0; i<n; i++) {
int cur=a[i][0];
int mycount=1;//记录相同元素个数
for (int j=1; j<m; j++) {
if(a[i][j]==cur)mycount++;
else{
cur=a[i][j];mycount=1;
}
if (mycount>=3) {
v.push_back(pair<int, int>(i,j) );
if(mycount==3){
v.push_back(pair<int, int>(i,j-1) );
v.push_back(pair<int, int>(i,j-2) );
}
}
}
}
for (int i=0; i<m; i++) {
int cur=a[0][i];
int mycount=1;//记录相同元素个数
for (int j=1; j<n; j++) {
if(a[j][i]==cur)mycount++;
else{
cur=a[j][i];mycount=1;
}
if (mycount>=3) {
v.push_back(pair<int, int>(j,i) );
if(mycount==3){
v.push_back(pair<int, int>(j-1,i) );
v.push_back(pair<int, int>(j-2,i) );
}
}
}
}
for (int i=0; i<v.size(); i++) {
a[v[i].first][v[i].second]=0;
}
for (int i=0; i<n; i++)
{
for (int j=0; j<m; j++) {
cout<<a[i][j]<<" ";
}
cout<<endl;
}
return 0;
}