【搜索+状态压缩】拯救大兵瑞恩

来源:互联网 发布:微信2级分销 源码下载 编辑:程序博客网 时间:2024/04/28 06:21

测试数据太坑了。找了一晚上没有找到错,最后发现是测试数据有问题。。。

拯救大兵瑞恩

(RESCUE.EXE)

1944年,特种兵麦克接到国防部的命令,要求立即赶赴太平洋上的一个孤岛,营救被敌军俘虏的大兵瑞恩。瑞恩被关押在一个迷宫里,迷宫地形复杂,但是幸好麦克得到了迷宫的地形图。

迷宫的外形是一个长方形,其在南北方向被划分为N行,在东西方向被划分为M列,于是整个迷宫被划分为N*M个单元。我们用一个有序数对(单元的行号,单元的列号)来表示单元位置。南北或东西方向相邻的两个单元之间可以互通,或者存在一扇锁着的门,又或者存在一堵不可逾越的墙。迷宫中有一些单元存放着钥匙,并且所有的门被分为P类,打开同一类的门的钥匙相同,打开不同类的门的钥匙不同。

大兵瑞恩被关押在迷宫的东南角,即(N,M)单元里,并已经昏迷。迷宫只有一个入口,在西北角,也就是说,麦克可以直接进入(1,1)单元。另外,麦克从一个单元移动到另一个相邻单元的时间为1,拿取所在单元的钥匙的时间以及用钥匙开门的时间忽略不计。

你的任务是帮助麦克以最快的方式抵达瑞恩所在单元,营救大兵瑞恩。


输入:

第一行是三个整数,依次表示N,M,P的值;

第二行是一个整数K,表示迷宫中门和墙的总个数;

第I+2行(1<=I<=K),有5个整数,依次为Xi1,Yi1,Xi2,Yi2,Gi:

当Gi>=1时,表示(Xi1,Yi1)单元与(Xi2,Yi2)单元之间有一扇第Gi类的门,当Gi=0时,表示(Xi1,Yi1)单元与(Xi2,Yi2)单元之间有一堵不可逾越的墙;

(其中,|Xi1-Xi2|+|Yi1-Yi2|=1,0<=Gi<=P)

第K+3行是一个整数S,表示迷宫中存放的钥匙总数;

第K+3+J行(1<=J<=S),有3个整数,依次为Xi1,Yi1,Qi:表示第J把钥匙存放在(Xi1,Yi1)单元里,并且第J把钥匙是用来开启第Qi类门的。(其中1<=Qi<=P)

注意:输入数据中同一行各相邻整数之间用一个空格分隔。

输出:

输出文件只包含一个整数T,表示麦克营救到大兵瑞恩的最短时间的值,若不存在可行的营救方案则输出-1。

输入输出示例:

输入文件

4 4 9

9

1 2 1 3 2

1 2 2 2 0

2 1 2 2 0

2 1 3 1 0

2 3 3 3 0

2 4 3 4 1

3 2 3 3 0

3 3 4 3 0

4 3 4 4 0

2

2 1 2

4 2 1

输出文件

14

参数设定:

3<=N,M<=15;

1<=P<=10;


这道题一开始有两个地方没有考虑到。

1、钥匙用过之后还可以再用 = =、

2、一个格子可能有多个钥匙= = 、

#include <cstdio>#define GETIND(a,b) (((a)-1)*m+((b)-1))#define KEYP(a) (1<<((a)-1))#define DELTA(a,b,c,d) ((a)==(c)?(((b)>(d)?2:3)):((a)>(c)?0:1))#include <set>long map[500][4];long key[500];long n;long m;long p;long s;long k;struct Sta{long ind;long sta;Sta():ind(0),sta(0){}Sta(long i,long s):ind(i),sta(s){}bool operator<(const Sta& st2)const {return ind<st2.ind || ind==st2.ind && sta < st2.sta;}};struct Qnode{Sta sta;long t;Qnode():t(0){}Qnode(long i,long s,long tt):sta(i,s),t(tt){}Qnode operator+(long dir){if (dir == 0){return Qnode(sta.ind-m,sta.sta,t+1);}else if (dir == 1){return Qnode(sta.ind+m,sta.sta,t+1);}else if (dir == 2){return Qnode(sta.ind-1,sta.sta,t+1);}else {return Qnode(sta.ind+1,sta.sta,t+1);}}Qnode operator|=(long pos){sta.sta |= pos;}Qnode operator!=(long pos){sta.sta -= pos;}long operator&(long pos){return sta.sta & pos;}bool end(){return sta.ind == GETIND(n,m);}};Qnode que[10000];std::set<Sta> vis;int main(){freopen("soldier.in","r",stdin);freopen("soldier.out","w",stdout);scanf("%ld%ld%ld%ld",&n,&m,&p,&k);for (long i=1;i<k+1;i++){long i1;long j1;long i2;long j2;long g;scanf("%ld%ld%ld%ld%ld",&i1,&j1,&i2,&j2,&g);if (g == 0) g = -1;map[GETIND(i1,j1)][DELTA(i1,j1,i2,j2)] = g;map[GETIND(i2,j2)][DELTA(i2,j2,i1,j1)] = g;}scanf("%ld",&s);for (long i=1;i<s+1;i++){long i1;long j1;long q;scanf("%ld%ld%ld",&i1,&j1,&q);key[GETIND(i1,j1)] |= KEYP(q);}for (long i=1;i<m+1;i++)map[GETIND(1,i)][0] = map[GETIND(n,i)][1] = -1;for (long i=1;i<n+1;i++)map[GETIND(i,1)][2] = map[GETIND(i,m)][3] = -1; long l = 0;long r = 0;Qnode u;Qnode v;que[++r] = u = Qnode(GETIND(1,1),0,0);vis.insert(u.sta);while (l<r){u = que[++l];for (long i=0;i<4;i++){if (map[u.sta.ind][i]==-1 || map[u.sta.ind][i] && !(u&KEYP(map[u.sta.ind][i])))continue;v = u + i;if (v.end()){printf("%ld",v.t);return 0;}if (key[v.sta.ind]) v|=key[v.sta.ind];//if (map[u.sta.ind][i]) v!=KEYP(map[u.sta.ind][i]);if (vis.find(v.sta) == vis.end()){vis.insert(v.sta);que[++r] = v;}}}printf("-1");return 0;}

原创粉丝点击