[蓝桥杯][2018年第九届真题]迷宫与陷阱（三维数组标记BFS）

题目描述
小明在玩一款迷宫游戏，在游戏中他要控制自己的角色离开一间由NxN个格子组成的2D迷宫。
小明的起始位置在左上角，他需要到达右下角的格子才能离开迷宫。
每一步，他可以移动到上下左右相邻的格子中（前提是目标格子可以经过）。
迷宫中有些格子小明可以经过，我们用’.‘表示；
有些格子是墙壁，小明不能经过，我们用’#'表示。
此外，有些格子上有陷阱，我们用’X’表示。除非小明处于无敌状态，否则不能经过。

有些格子上有无敌道具，我们用’%'表示。
当小明第一次到达该格子时，自动获得无敌状态，无敌状态会持续K步。
之后如果再次到达该格子不会获得无敌状态了。

处于无敌状态时，可以经过有陷阱的格子，但是不会拆除/毁坏陷阱，即陷阱仍会阻止没有无敌状态的角色经过。

给定迷宫，请你计算小明最少经过几步可以离开迷宫
输入
第一行包含两个整数N和K。 (1 <= N <= 1000 1 <= K <= 10)
以下N行包含一个NxN的矩阵。
矩阵保证左上角和右下角是’.’。
输出
一个整数表示答案。如果小明不能离开迷宫，输出-1。
样例输入
5 3
…XX
##%#.
…#.
.###.
…
样例输出
10
思路：挺明显的一个bfs题目，但是某个点不能单纯的走过就不能走了，因为有另外一个元素k。而且k很小。所以我们就用一个三维数组标记，如果某个点在无敌状态为k时已经走过了，就不能再走了。剩下的就是在每一种状态时的判断了。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int maxx=1e3+100;
struct node{
	int x,y,k,val;
	node(int a,int b,int c,int v)
	{
		x=a,y=b,k=c,val=v;
	}
};
int d[][2]={{1,0},{0,1},{0,-1},{-1,0}};
char s[maxx][maxx];
int vis[maxx][maxx][11];
int n,k;

inline int bfs()
{
	queue<node> q;
	vis[0][0][0]=1;
	q.push(node(0,0,0,0));
	while(q.size())
	{
		node u=q.front();
		q.pop();
		if(u.x==n-1&&u.y==n-1) return u.val;
		for(int i=0;i<4;i++)
		{
			int tx=u.x+d[i][0];
			int ty=u.y+d[i][1];
			if(tx<0||ty<0||tx>=n||ty>=n||s[tx][ty]=='#') continue;
			if(s[tx][ty]=='%'&&vis[tx][ty][k]==0)
			{
				vis[tx][ty][k]=1;
				q.push(node(tx,ty,k,u.val+1));
			}
			else 
			{
				if(s[tx][ty]=='X'&&u.k&&vis[tx][ty][u.k-1]==0)
				{
					vis[tx][ty][u.k-1]=1;
					q.push(node(tx,ty,u.k-1,u.val+1));
				}
				else if(s[tx][ty]=='.'&&u.k&&vis[tx][ty][u.k-1]==0)
				{
					vis[tx][ty][u.k-1]=1;
					q.push(node(tx,ty,u.k-1,u.val+1));
				}
				else if(s[tx][ty]=='.'&&u.k==0&&vis[tx][ty][0]==0)
				{
					vis[tx][ty][0]=1;
					q.push(node(tx,ty,0,u.val+1));
				}
			}
		}
	}
	return -1;
}
int main()
{
	scanf("%d%d",&n,&k);
	memset(vis,0,sizeof(vis));
	for(int i=0;i<n;i++) scanf("%s",s[i]);
	int ans=bfs();
	if(ans==-1) printf("-1\n");
	else printf("%d\n",ans);
	return 0;
}

努力加油a啊，(^o)/~