2017模拟赛 密室（多维最短路）

二维最短路记录 表示当前状态的最小情况
注意初始情况

代码

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
#define maxn 7001
#define inf 1e9
int n,m,k;
int End[maxn],Next[maxn],Last[maxn],Len[maxn];
int cnt;
int a[5005];
void insert(int x,int y,int z){
    Next[++cnt]=Last[x];
    Last[x]=cnt;
    End[cnt]=y;
    Len[cnt]=z;
}
int minn=inf;
struct node{
    int x,now;
};
queue<node>q;
bool mark[maxn][1025];
int dis[maxn][1025];
void spfa(){
    int i,j,k;
    for(i=1;i<=n;i++) 
      for(j=0;j<=1024;j++)
         dis[i][j]=inf;
    dis[1][a[1]]=0;
    node t;
    t.x=1;t.now=a[1];
    q.push(t);
    while(q.size()){
        t=q.front();
        q.pop();
        int x=t.x,now=t.now;
        mark[x][now]=false;
        for(i=Last[x];i;i=Next[i]){
            if((now&Len[i])!=Len[i]) continue;
            int en=End[i];
            if(dis[en][now|a[en]]>dis[x][now]+1){
                dis[en][now|a[en]]=dis[x][now]+1;
                if(mark[en][now|a[en]]==false){
                    node tmp;
                    tmp.x=en;
                    tmp.now=(now|a[en]);
                    q.push(tmp);
                    mark[en][now|a[en]]=true;
                }
            }
        }
    }
}
int main()
{
//  freopen("room.in","r",stdin);
//  freopen("room.out","w",stdout);
    int i,j,k;
    scanf("%d%d%d",&n,&m,&k);
    for(i=1;i<=n;i++){
        int x;
        for(j=1;j<=k;j++){
            scanf("%d",&x);
            a[i]=a[i]|(x<<(j-1));
        }
    }
    for(i=1;i<=m;i++){
        int x,y,z;
        int len=0;
        scanf("%d%d",&x,&y);
        for(j=1;j<=k;j++){
            scanf("%d",&z);
            len=len|(z<<(j-1));
        }
        insert(x,y,len);
    }
    spfa();
    int minn=inf;
    for(i=0;i<=1024;i++) minn=min(dis[n][i],minn);
    if(minn==inf) cout<<"No Solution";
    else cout<<minn;
}