洛谷 [P4011] 孤岛营救问题

状压+BFS

通过观察数据范围可知,我们应该状压钥匙种类,直接BFS即可

注意,一个点处可能不知有一把钥匙

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <queue>
using namespace std;
bool f[14][14][1200];
int n,m,p,k,s,key[15][15],block1[15][15],block2[15][15],dx[4]={-1,0,0,1},dy[4]={0,-1,1,0};
struct pa{
	int x,y,key,tim;
};
queue<pa> q;
int main(){
	cin>>n>>m>>p;
	cin>>k;
	for(int i=1;i<=k;i++){
		int a1,a2,b1,b2,g;
		cin>>a1>>b1>>a2>>b2>>g;
		if(!g) g=50;
		if(a1==a2) block1[a1][min(b1,b2)]=g;
		else block2[min(a1,a2)][b1]=g;
	}
	cin>>s;
	for(int i=1;i<=s;i++){
		int a,b,ty;
		cin>>a>>b>>ty;
		key[a][b]=key[a][b]|(1<<(ty-1));
	}
/*	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			printf("%d ",block2[i][j]);
		}
		printf("\n");
	}*/
	pa a;
	a.x=1;a.y=1;a.key=key[1][1];a.tim=0;
	q.push(a);f[1][1][a.key]=1;
	while(!q.empty()){
		pa u=q.front();q.pop();
		//printf("%d %d %d u\n",u.x,u.y,u.key);
		for(int i=0;i<4;i++){
			int x=u.x+dx[i],y=u.y+dy[i];
			if(x>0&&x<=n&&y>0&&y<=m){
				if(((i==1)||(i==2))&&block1[min(x,u.x)][min(y,u.y)]){
					if(block1[min(x,u.x)][min(y,u.y)]==50) continue;
					if(!((u.key>>(block1[min(x,u.x)][min(y,u.y)]-1))&1)) continue;
				}
				if(((i==0)||(i==3))&&block2[min(x,u.x)][min(y,u.y)]){
					if(block2[min(x,u.x)][min(y,u.y)]==50) continue;
					if(!((u.key>>(block2[min(x,u.x)][min(y,u.y)]-1))&1)) continue;
				}
				pa v;
				v.x=x;v.y=y;v.key=u.key|key[x][y];v.tim=u.tim+1;
				if(!f[x][y][v.key]) q.push(v),f[x][y][v.key]=1;
				if(x==n&&y==m){
					printf("%d\n",v.tim);
					return 0;
				}
			}
		}
	}
	printf("-1\n");
	return 0;
}