CCF认证 2015-12 消除类游戏

用一个数组先标记要消除的方块，最后统一消除，这样可以处理行列同时被消除的情况

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=30+2;
const int M=30+2;
int a[N][M],n,m;
bool b[N][M]; //用来标记 
int main()
{
	memset(b,false,sizeof(b));
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	for(int j=0;j<m;j++)
	scanf("%d",&a[i][j]);
	for(int i=0;i<n;i++) //标记列消除 
	for(int j=1;j<m-1;j++)
	if(a[i][j-1]==a[i][j]&&a[i][j+1]==a[i][j])
	b[i][j-1]=b[i][j]=b[i][j+1]=true;
	for(int j=0;j<m;j++) //标记行消除 
	for(int i=1;i<n-1;i++)
	if(a[i-1][j]==a[i][j]&&a[i+1][j]==a[i][j])
	b[i-1][j]=b[i][j]=b[i+1][j]=true;
	for(int i=0;i<n;i++) //消除 
	for(int j=0;j<m;j++)
	if(b[i][j]) a[i][j]=0;
	for(int i=0;i<n;i++) //输出 
	{
		for(int j=0;j<m-1;j++)
		printf("%d ",a[i][j]);
		printf("%d\n",a[i][m-1]);
	}
	return 0;
}