loj6121「网络流 24 题」孤岛营救问题（状态压缩+spfa）

又是一道网络流24题中的最短路x
读题读了半天才读懂，我可能傻掉了。可以状压分层图来跑最短路（把 点(x,y)时拥有钥匙状态S压成一个点来跑spfa）。点数应该是 O(nm2p) 的。可以最后连向一个超级汇，方便得到答案。

#include 
#include 
#include 
#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 110000
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,p,id[11][11][1030],tot=0,key[11][11],T=102401,bin[20],door[110][110];
int dx[]={0,0,1,-1},dy[]={1,-1,0,0},h[N],num=1,dis[N];
bool inq[N];
struct edge{
    int to,next,val;
}data[N<<2];
inline void add(int x,int y,int val){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
}
inline void spfa(){
    queue<int>q;memset(dis,inf,sizeof(dis));
    q.push(1);dis[1]=0;inq[1]=1;
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;
            if(dis[x]+data[i].valif(!inq[y]) inq[y]=1,q.push(y);
            }
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();p=read();bin[0]=1;
    for(int i=1;i<=p;++i) bin[i]=bin[i-1]<<1;
    for(int i=0;i<=bin[p]-1;++i)
        for(int x=1;x<=n;++x)
            for(int y=1;y<=m;++y) id[x][y][i]=++tot;
    int owo=read();
    while(owo--){
        int x1=read(),y1=read(),x2=read(),y2=read(),val=read();
        door[id[x1][y1][0]][id[x2][y2][0]]=door[id[x2][y2][0]][id[x1][y1][0]]=val?val:-1;
    }owo=read();
    while(owo--){
        int x=read(),y=read();key[x][y]|=bin[read()-1];
    }
    for(int i=0;i<=bin[p]-1;++i)
        for(int x=1;x<=n;++x)
            for(int y=1;y<=m;++y)
                for(int k=0;k<4;++k){
                    int xx=x+dx[k],yy=y+dy[k];
                    if(xx<1||xx>n||yy<1||yy>m) continue;
                    int op=door[id[x][y][0]][id[xx][yy][0]];
                    if(op==-1||op&&!(i&bin[op-1])) continue;
                    add(id[x][y][i],id[xx][yy][i|key[xx][yy]],1);
                }
    for(int i=0;i<=bin[p]-1;++i) add(id[n][m][i],T,0);
    spfa();if(dis[T]==inf) puts("-1");
    else printf("%d\n",dis[T]);
    return 0;
}