洛谷P1126 机器人搬重物(bfs)

传送门 难度
https://www.luogu.com.cn/problem/P1126 普及+/提高

这是一道bfs的水题

分析

这道题用一下bfs就可以,难就难在如何进行表示

  • 方向:0,1,2,3来表示上右下左,转向时用(+1)%4或(+3)%4就能实现右转、左转
  • 机器人位置:机器人的点是在网格上,表示的时候以机器人的左上角的点作为位置,判断的时候以该点为基准点进行行动的模拟
  • 为了使机器人在同一位置同一方向进行bfs的时候不重复,使用一个hash函数进行映射

AC代码

#include
#include
#include
#include
#include

using namespace std;

int mp[55][55];
bool flag[400000];
int N, M;
struct Node {
	int x, y;
	int d;
	int t;
};
queue<Node> q;
int st[4][2] = { {-1,0},{0,1},{1,0},{0,-1} };
int sx, sy, dx, dy;
int f;

bool check(int x, int y) {
	if (mp[x][y] || mp[x][y + 1] || mp[x + 1][y] || mp[x + 1][y + 1])
		return true;
	return false;
}

bool mark(int x,int y,int d) {//hash函数,保证不会重复就可以了,数字随意
	int xy = 5000 * x + 50 * y + d;
	if (flag[xy])
		return true;
	flag[xy] = true;
	return false;
}

int bfs() {
	q.push({ sx,sy,f,0 });
	while (!q.empty()) {
		Node n = q.front();
		q.pop();
		int xx = n.x, yy = n.y, dd = n.d, tt = n.t;
		if (xx == dx && yy == dy)
			return tt;
		if (mark(xx,yy,dd))
			continue;
		tt++;
		//cout << xx << yy << dd << endl;
		q.push({ xx,yy,(dd + 3) % 4,tt });//左转
		q.push({ xx,yy,(dd + 1) % 4,tt });//右转
		int temx, temy;
		for (int i = 1; i <= 3; ++i) {//前进1、2、3步
			temx = xx + st[dd][0] * i;
			temy = yy + st[dd][1] * i;
			if (temx<0 || temy<0 || temx>(N - 2) || temy>(M - 2) || check(temx, temy))
				break;
			q.push({temx, temy, dd, tt});
		}
	}
	return -1;
}


int main() {
	scanf("%d%d", &N, &M);
	for (int i = 0; i < N; ++i)
		for (int j = 0; j < M; ++j)
			scanf("%d", &mp[i][j]);
	cin >> sx >> sy >> dx >> dy;
	sx--; sy--; dx--; dy--;
	char c;
	cin >> c;
	switch (c)
	{
		case 'N': {f = 0; break; }
		case 'E': {f = 1; break; }
		case 'S': {f = 2; break; }
		case 'W': {f = 3; break; }
	}
	cout << bfs() << endl;
	return 0;
}

你可能感兴趣的:(BFS)