【搜索】NOIP2011Mayan游戏

题目链接：洛谷1312

题目描述

Mayan Puzzle是最近流行起来的一个游戏。游戏界面是一个7行×5列的棋盘，上面堆放着一些方块，方块不能悬空堆放，即方块必须放在最下面一行，或者放在其他方块之上。游戏通关是指在规定的步数内消除所有的方块，消除方块的规则如下：

1 、每步移动可以且仅可以沿横向（即向左或向右）拖动某一方块一格：当拖动这一方块时，如果拖动后到达的位置（以下称目标位置）也有方块，那么这两个方块将交换位置（参见输入输出样例说明中的图6到图7）；如果目标位置上没有方块，那么被拖动的方块将从原来的竖列中抽出，并从目标位置上掉落（直到不悬空，参见下面图1和图2）；

2 、任一时刻，如果在一横行或者竖列上有连续三个或者三个以上相同颜色的方块，则它们将立即被消除（参见图1到图3）。

注意：

a) 如果同时有多组方块满足消除条件，几组方块会同时被消除（例如下面图4，三个颜色为1 的方块和三个颜色为 2 的方块会同时被消除，最后剩下一个颜色为2的方块）。

b) 当出现行和列都满足消除条件且行列共享某个方块时，行和列上满足消除条件的所有方块会被同时消除（例如下面图5所示的情形，5个方块会同时被消除）。

3 、方块消除之后，消除位置之上的方块将掉落，掉落后可能会引起新的方块消除。注意：掉落的过程中将不会有方块的消除。

上面图1到图3给出了在棋盘上移动一块方块之后棋盘的变化。棋盘的左下角方块的坐标为（0, 0），将位于（3, 3）的方块向左移动之后，游戏界面从图1变成图2所示的状态，此时在一竖列上有连续三块颜色为4的方块，满足消除条件，消除连续3块颜色为4的方块后，上方的颜色为3的方块掉落，形成图3所示的局面。

输入输出格式

输入格式：

共 6 行。

第一行为一个正整数n，表示要求游戏通关的步数。

接下来的5行，描述7×5 的游戏界面。每行若干个整数，每两个整数之间用一个空格隔开，每行以一个0结束，自下向上表示每竖列方块的颜色编号（颜色不多于10种，从1开始顺序编号，相同数字表示相同颜色）。

输入数据保证初始棋盘中没有可以消除的方块。

输出格式：

如果有解决方案，输出n行，每行包含3个整数x,y,g，表示一次移动，每两个整数之间用一个空格隔开，其中(x ,y)表示要移动的方块的坐标，g 表示移动的方向，1 表示向右移动，−1表示向左移动。注意：多组解时，按照x为第一关健字，y为第二关健字，1优先于−1 ，给出一组字典序最小的解。游戏界面左下角的坐标为(0 ,0)。

如果没有解决方案，输出一行，包含一个整数−1。

输入输出样例

输入样例#1： 

3
1 0
2 1 0
2 3 4 0
3 1 0
2 4 3 4 0

输出样例#1： 

2 1 1
3 1 1
3 0 1

说明

【输入输出样例说明】

按箭头方向的顺序分别为图6到图11

样例输入的游戏局面如上面第一个图片所示，依次移动的三步是：(2 ,1)处的方格向右移动，(3,1)处的方格向右移动，(3,0)处的方格向右移动，最后可以将棋盘上所有方块消除。

【数据范围】

对于30%的数据，初始棋盘上的方块都在棋盘的最下面一行；

对于100%的数据，0<n≤5 。

NOIP2011提高组day1第3题

题解：

首先观察范围和初步分析可确定本题应用搜索算法，因为题目中步数小于5，范围很小，因而考虑用DFS而非BFS。而还要保证步骤字典序最小，因此选择从左到右，从下到上的搜索顺序，同时发现，方块左移和方块右移可以在特殊记录空方块右移（在输出答案时特殊考虑一下）后统一考虑为方块右移，因而搜索顺序为：从左到右，从上到下，只考虑方块右移。

然后就可以开始搜索了，不过本题还有一些比较棘手的步骤，即方块的下落和消除，以及小细节：搜索的优化。

考虑方块下落。我们可以把同一列的方块先收集到一个队列中，清空该列，然后在将队列中的元素弹出，顺次放在棋盘的该列即可。这样下落问题就解决了。再考虑方块的消除，暴力的消除会比较麻烦，因此想到用一个标记数组来辅助消除。每次以一个位置为起点，搜索横三和纵三的方块，在标记数组中打标记，就可以解决在消除中方块公用的问题了。

最后是搜索的优化：

第一、我们已经从搜索顺序上进行了优化，找到的第一个解就是字典序最小的解；

第二、如果一个状态中某一种方块的个数大于0而小于三，则该方块永远不会被消除，可以判定当前状态失败，直接返回上一层；

第三、在考虑方块的移动时，只需考虑颜色不同的方块的交换。

到此，所有的步骤就都清楚了，总结一下，本题需要的不仅仅是搜索，还有在处理模拟过程中的一些技巧，还有搜索的剪枝，是不可多得的好题。

代码如下。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=10;
int n;
int state[maxn][maxn];
struct Answer{
	int x,y;
	bool flag;//flag=true表示方块向左移动,如此定义达到剪枝目的.
}ans[maxn];
bool Empty(){//判断棋盘是否为空.
	for (int i=0;i<5;i++)
		for (int j=0;j<7;j++)
			if (state[i][j]) return false;
	return true;
}
void Drop(){//方块下落.
	queue <int> Q;
	for (int i=0;i<5;i++){
		int cnt=0;
		for (int j=0;j<7;j++)
			if (state[i][j]) Q.push(state[i][j]);//先把所有方块收集到一个队列里.
		memset(state[i],0,sizeof(state[i]));//清空该列.
		while (!Q.empty()) state[i][cnt++]=Q.front(),Q.pop();//将队列中的元素弹出放到棋盘中.
	}
}
bool Clear(){
	bool CanClear;//能否消除标记.
	bool mark[maxn][maxn];
	memset(mark,false,sizeof(mark));
	for (int i=0;i<5;i++){
		for (int j=0;j<7;j++){
			if (state[i][j]==0) continue;//没有方块.
			int tempx;
			for (tempx=i;tempx+1<5 && state[tempx+1][j]==state[i][j];tempx++);
			//用tempx找到横向能消除的最右位置.
			if (tempx-i>=2){//如果(横向)相连的方块长度超过3,则能消除.
				for (int tx=i;tx<=tempx;tx++) mark[tx][j]=true;//在mark数组对应的位置上打上标记.
				CanClear=true;//能消除.
			}
			int tempy;
			for (tempy=j;tempy+1<7 && state[i][tempy+1]==state[i][j];tempy++);
			//用tempy找到纵向能找到的最上位置.
			if (tempy-j>=2){//如果(纵向)相连的方块长度超过3,则能消除.
				for (int ty=j;ty<=tempy;ty++) mark[i][ty]=true;//在mark数组对应的位置上打上标记.
				CanClear=true;//能消除.
			}
		}
	}
	for (int i=0;i<5;i++)
		for (int j=0;j<7;j++)
			if (mark[i][j]) state[i][j]=0;//将打上标记的方块全部消除.
	return CanClear;//返回是否能消除.
}
void DFS(int depth){
	if (depth>n){//操作数用尽.
		if (Empty()){//如果为空则输出操作序列.
			for (int i=1;i<=n;i++)
				if (ans[i].flag) printf("%d %d %d\n",ans[i].x+1,ans[i].y,-1);
				else printf("%d %d %d\n",ans[i].x,ans[i].y,1);
			exit(0);//直接结束程序.
		}
		return ;
	}
	int sum[maxn*2]={0};
	//统计每个颜色的方块的数量.
	for (int i=0;i<5;i++)
		for (int j=0;j<7;j++) sum[state[i][j]]++;
	//剪枝--如果某方块的数量小于三,则该种方块一定不可以被消除,失败.
	for (int i=1;i<=15;i++)
		if (sum[i]!=0 && sum[i]<3) return ;
	for (int i=0;i<4;i++)//这种搜索顺序能够保证搜到的第一种方案就是字典序最小的.
		for (int j=0;j<7;j++){
			if (state[i][j]==state[i+1][j]) continue;//剪枝--如果颜色相同,则无需交换.
			ans[depth].x=i,ans[depth].y=j,ans[depth].flag=(!state[i][j]);//记录当前操作.
			int temp[maxn][maxn];
			memcpy(temp,state,sizeof(temp));//将当前状态存到临时数组中,防止丢失.
			swap(state[i][j],state[i+1][j]);//进行操作(交换).
			Drop();//先掉落.
			while (Clear()) Drop();//反复消除、掉落.
			DFS(depth+1);//迭代搜索.
			ans[depth].x=0,ans[depth].y=0,ans[depth].flag=0;//清空当前操作.
			memcpy(state,temp,sizeof(state));//还原当前状态.
		}
}
int main(){
//	freopen("Mayan.in","r",stdin);
//	freopen("Mayan.out","w",stdout);
	scanf("%d",&n);
	for (int i=0;i<5;i++)
		for (int j=0;;j++){
			scanf("%d",&state[i][j]);
			if (state[i][j]==0) break;
		}
	DFS(1);
	printf("-1\n");//无解情况.
	return 0;
}