【Week 12 CSP-M3 B】消消乐大师-Q老师

题目描述

输入输出格式

样例

数据规模

思路

使用二维数组记录原始棋盘，数组基本元素为pair<int ,bool>，int变量用来记录原始棋子，bool变量用来记录是否可以删除，初始化为false。
遍历数组，对于每一个棋子，分别向左右，上下进行延伸，若相同棋子数大于三，则再重新延伸一次，并置bool变量为true。
最后根据每个棋子的bool变量判断输出棋子原值还是输出0即可。

代码

#include <iostream>
#include <stdio.h>
using namespace std;
const int size=30+10;
pair<int,bool> num[size][size];
int n,m;
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};
void solve()
{
	int count;bool flag1,flag2;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			flag1=false;flag2=false;count=0;
			int color=num[i][j].first,y=i,x=j;
			while(y+dy[0]<=n&&num[y+dy[0]][x].first==color)
			{
				y=y+dy[0];count++;
			}
			y=i;x=j;
			while(y+dy[1]>=1&&num[y+dy[1]][x].first==color)
			{
				y=y+dy[1];count++;
			}
			if(count>=2)flag1=true;
			y=i;x=j;count=0;
			while(x+dx[2]<=m&&num[y][x+dx[2]].first==color)
			{
				x+=dx[2];count++;
			}
			y=i;x=j;
			while(x+dx[3]>=1&&num[y][x+dx[3]].first==color)
			{
				x+=dx[3];count++;
			}
			if(count>=2)flag2=true;
			if(flag1)
			{
				num[i][j].second=true;
				y=i;x=j;
				while(y+dy[1]>=1&&num[y+dy[1]][x].first==color)
				{
					y=y+dy[1];num[y][x].second=true;
				}
				y=i;x=j;
				while(y+dy[0]<=n&&num[y+dy[0]][x].first==color)
				{
					y=y+dy[0];num[y][x].second=true;
				}
			}
			if(flag2)
			{
				num[i][j].second=true;
				y=i;x=j;
				while(x+dx[2]<=m&&num[y][x+dx[2]].first==color)
				{
					x+=dx[2];num[y][x].second=true;
				}
				y=i;x=j;
				while(x+dx[3]>=1&&num[y][x+dx[3]].first==color)
				{
					x+=dx[3];num[y][x].second=true;
				}
			}
		}
	}
}
int main(int argc, char** argv) {
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&num[i][j].first);
			num[i][j].second=false;			
		}
	}	
	solve();
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(j==1)
			{
				if(num[i][j].second)printf("0");
				else printf("%d",num[i][j].first);
			}
			else
			{
				if(num[i][j].second)printf(" 0");
				else printf(" %d",num[i][j].first);
			}
		}
		printf("\n");
	}
	return 0;
}