CSP模拟3——B

消消乐大师——Q老师

Q老师是个很老实的老师，最近在积极准备考研。Q老师平时只喜欢用Linux系统，所以Q老师的电 脑上没什么娱乐的游戏，所以Q老师平时除了玩Linux上的赛车游戏SuperTuxKart之外，就是喜欢 消消乐了。 游戏在一个包含有 n 行 m 列的棋盘上进行，棋盘的每个格子都有一种颜色的棋子。当一行或一列 上有连续三个或更多的相同颜色的棋子时，这些棋子都被消除。当有多处可以被消除时，这些地 方的棋子将同时被消除。 一个棋子可能在某一行和某一列同时被消除。 由于这个游戏是闯关制，而且有时间限制，当Q老师打开下一关时，Q老师的好哥们叫Q老师去爬 泰山去了，Q老师不想输在这一关，所以它来求助你了！！
输入描述

输入第一行包含两个整数n,m，表示行数和列数 接下来n行m列，每行中数字用空格隔开，每个数字代表这个位置的棋子的颜色。数字都大于0.

输出描述

输出n行m列，每行中数字用空格隔开，输出消除之后的棋盘。（如果一个方格中的棋子被消除， 则对应的方格输出0，否则输出棋子的颜色编号。）

样例输入1

4 5 
2 2 3 1 2 
3 4 5 1 4 
2 3 2 1 3 
2 2 2 4 4

样例输出1

2 2 3 0 2 
3 4 5 0 4 
2 3 2 0 3 
0 0 0 4 4

样例输入2

4 5 
2 2 3 1 2 
3 1 1 1 1 
2 3 2 1 3 
2 2 3 3 3

样例输出2

2 2 3 0 2 
3 0 0 0 0 
2 3 2 0 3 
2 2 0 0 0

问题分析

一开始想的是BFS对上下左右四个方向遍历，但是太麻烦了。再看两遍样例想到消除只发生在行或列有三个以上相同字符。
因此可以直接遍历行，每遇到三个一样的就标记一下。在按照列遍历一次，这样就可以标记出所有能消除的。对于样例二这种有重合的标记两次对结果也不会产生影响。

注意：输出格式行末没有空格

代码实现

#include<iostream>
#include<string.h>
using namespace std;

int n,m;
int a[35][35];
int mark[35][35];

int main()
{
	cin>>n>>m;
	memset(mark, 0,sizeof(mark));
	for(int i=1; i<=n; i++)
	{
		for(int j=1; j<=m; j++)
		{
			cin>>a[i][j];
		}
	}
	
	for(int i=1; i<=n; i++)
	{
		for(int j=2; j<m; j++)
		{
			if(a[i][j] == a[i][j-1] && a[i][j]==a[i][j+1])
			    mark[i][j]=mark[i][j-1]=mark[i][j+1]=1;
		}
	}
	
	for(int j=1; j<=m; j++)
	{
		for(int i=2; i<n; i++)
		{
			if(a[i][j]==a[i-1][j] && a[i][j]==a[i+1][j])
			    mark[i][j]=mark[i-1][j]=mark[i+1][j]=1;
		}
	}
	
	for(int i=1; i<=n; i++)
	{
		for(int j=1; j<m; j++)
		{
			if(mark[i][j]==0)
			    cout<<a[i][j]<<" ";
			else
			    cout<<"0 ";
		}
		if(mark[i][m]==0)
			cout<<a[i][m]<<endl;
		else
			cout<<"0"<<endl;
	}
	
	return 0;
}