洛谷 [P4011] 孤岛营救问题

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