习题6-5 UVa1600 Patrol Robot(BFS)

题意：

机器人走迷宫求最小步数，不同之处在于机器人最多能一次连续穿过k堵墙。

要点：

BFS模板题稍微有些变换，将vis数组增加到三维来记录穿墙数即可，因为有的时候就算坐标相同，但穿过墙数不同也是不同的。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
using namespace std;
int map[50][50],vis[50][50][50];//visit多一维来记录穿墙数
int m, n, k;
int d[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
struct node
{
	int x, y;
	int count;
	int layer;//用layer记录穿墙数
};

int bfs(int x,int y)
{
	memset(vis, 0, sizeof(vis));
	queue<node> q;
	node c;
	c.x = x; c.y = y;
	c.count = 0;c.layer=0;
	q.push(c);
	vis[x][y][0] = 1;
	while (!q.empty())
	{
		node c = q.front(); q.pop();
		if (c.x == m&&c.y == n)
			return c.count;
		for (int i = 0; i < 4; i++)
		{
			node v;
			v.x = c.x + d[i][0];
			v.y = c.y + d[i][1];
			v.layer = c.layer;
			if (map[v.x][v.y])
				v.layer++;     //遇到墙则穿墙数+1
			else
				v.layer = 0;	//如果不穿墙，穿墙数归零
			v.count = c.count + 1;
			if (v.layer<=k&&!vis[v.x][v.y][v.layer]&&v.x>=1&&v.x<=m&&v.y>=1&&v.y<=n)
			{
				vis[v.x][v.y][v.layer] = 1;//这里多一维的目的是有时候坐标相同但穿墙数不同也是能经过的
				q.push(v);
			}
		}
	}
	return -1;
}

int main()
{
	int t;
	scanf("%d", &t);
	while (t--)
	{
		int i,j;
		scanf("%d%d", &m, &n);
		scanf("%d", &k);
		for (i = 1; i <= m; i++)
			for (j = 1; j <= n; j++)
				scanf("%d", &map[i][j]);
		printf("%d\n", bfs(1,1));
	}
	return 0;
}