hdu4725 The Shortest Path in Nya Graph

这道题看了下很多人都是把每一层拆成两个点然后建图做的。

我的思路很直接，也不用建图，直接在更新每个点时更新他相邻的边和相邻的层，当然前提是每个点只更新一次，每个层也只更新一次，这样才能确保时间复杂度。

这里我用了两个邻接表，一个是邻接边，一个是邻接层，最后用优先队列优化下。

下面是代码

#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
const int M=2*111111;
int fir[M],u[M],v[M],w[M],nxt[M],e;//边的邻接
int n,m,c;
int lfir[M],lu[M],lnxt[M],le;//层的邻接
int lay[M];//表示每个点所在的层
int dis[M],vis[M];//dis表示与起点距离，vis标记边是否访问过
int lvis[M];//表示层是否访问过
struct cmp
{
    bool operator()(int a,int  b)
    {
        return dis[a]>dis[b];
    }
};

priority_queue<int,vector<int>,cmp>Q;
void linsert(int la,int num)//插入层
{
    lu[le]=num;
    lnxt[le]=lfir[la];
    lfir[la]=le++;
}
void insert(int a,int b,int c)//插入边
{
    u[e]=a;v[e]=b;w[e]=c;
    nxt[e]=fir[a];
    fir[a]=e++;
}
const int inf=2000000000;
int bfs()
{
    int i,j,k,tm;
    for(i=1;i<=n;i++)dis[i]=(i==1?0:inf);
    memset(vis,0,sizeof(vis));
    memset(lvis,0,sizeof(lvis));
    while(!Q.empty())Q.pop();
    Q.push(1);
    while(!Q.empty())
    {
        int p=Q.top();Q.pop();
        if(p==n)break;
        if(vis[p])continue;
        vis[p]=1;
        for(k=fir[p];k!=-1;k=nxt[k])if(dis[ v[k] ]>dis[p]+w[k]){//更新边
            dis[ v[k] ]=dis[p]+w[k];
            Q.push(v[k]);
        }

        if(lvis[lay[p]])continue;
        lvis[lay[p]]=1;

        tm=lay[p]-1;
        for(k=lfir[tm];k!=-1;k=lnxt[k])if(dis[ lu[k] ]>dis[p]+c){//更新上一层
            dis[ lu[k] ]=dis[p]+c;
            Q.push(lu[k]);
        }

        tm=lay[p]+1;
         for(k=lfir[tm];k!=-1;k=lnxt[k])if(dis[ lu[k] ]>dis[p]+c){//更新下一层
            dis[ lu[k] ]=dis[p]+c;
            Q.push(lu[k]);
        }
    }
    if(dis[n]==inf)return -1;
    return dis[n];
}
int main()
{
    int t,i,j,k,cas=1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&c);
        le=0;
        memset(lfir,-1,sizeof(lfir));
        for(i=1;i<=n;i++){
            scanf("%d",&lay[i]);
            linsert(lay[i],i);
        }
        int a,b,c;
        e=0;
        memset(fir,-1,sizeof(fir));
        for(i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&c);
            insert(a,b,c);
            insert(b,a,c);
        }
        int ans=bfs();
        printf("Case #%d: %d\n",cas++,ans);
    }
    return 0;
}