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

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

拯救大兵瑞恩

(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;
}