Airport Express UVA

Airport Express

 UVA - 11374

题意：n个点，有m条普通路径，k条高速路径，但是k条只能选一条走。问从s到e最短时间。

如果选a-->b这条高速，那么s-->a和b--->e必然也要是最短路。

于是我们可以先用两次dijkstra预处理出s到各点的最短路和e到各点的最短路，然后枚举k条高速走哪条。

输出路径的时候，可以递归~

#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxv=1010;
const int maxe=1010;

struct Edge
{
    int u,v,w;
    int nex;
}e[maxe<<1];
int head[maxv];
int cnt=0;
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0;
}
void add(int u,int v,int w)
{
    e[cnt].u=u;
    e[cnt].v=v;
    e[cnt].w=w;
    e[cnt].nex=head[u];
    head[u]=cnt++;
}

typedef pair<int,int> PII;
int dis[maxv];
int par[maxv];
void dijkstra(int s)
{
    priority_queue<PII,vector<PII>,greater<PII> > pq;
    for(int i=0;i<maxv;i++) dis[i]=inf;
    dis[s]=0;
    pq.push(PII(0,s));
    while(!pq.empty())
    {
        PII temp=pq.top();
        pq.pop();
        int u=temp.second;
        if(dis[u]<temp.first) continue;
        for(int i=head[u];i!=-1;i=e[i].nex)
        {
            if(dis[e[i].v]>dis[u]+e[i].w)
            {
                dis[e[i].v]=dis[u]+e[i].w;
                par[e[i].v]=i;
                pq.push(PII(dis[e[i].v],e[i].v));
            }
        }
    }
}
int n,ss,ee;
int ds[maxv],de[maxv];
int p[maxv],p1[maxv];

void print(int a)
{
    if(a==ss)
    {
        printf("%d",a+1);
    }
    else
    {
        int x=p1[a];
        print(e[x].u);
        printf(" %d",a+1);
    }
    return ;
}

int main()
{
    int kase=0;

    while(scanf("%d%d%d",&n,&ss,&ee)!=EOF&&n)
    {
        if(kase++) puts("");
        ss--;ee--;
        init();
        int u,v,w;
        int m;
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            u--,v--;
            add(u,v,w);
            add(v,u,w);
        }
        dijkstra(ss);
        for(int i=0;i<n;i++) {ds[i]=dis[i];p1[i]=par[i];}
        pa[ss]=-1;
        dijkstra(ee);
        for(int i=0;i<n;i++) {de[i]=dis[i];p[i]=par[i];}
        p[ee]=-1;


        int ans1=ds[ee];
        int a=-1,b=-1;
        int k;
        scanf("%d",&k);
        for(int i=0;i<k;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            u--;v--;
            if(ans1>ds[u]+w+de[v])
            {
                ans1=ds[u]+w+de[v];
                a=u,b=v;
            }
            if(ans1>ds[v]+w+de[u])
            {
                ans1=ds[v]+w+de[u];
                a=v,b=u;
            }
        }

        int flag=0;
        if(a==-1)
        {
            int x=ss;
            while(x!=ee)
            {
                if(!flag){printf("%d",x+1);flag=1;}
                else printf(" %d",x+1);
                x=e[p[x]].u;
            }
            printf(" %d
",ee+1);
            printf("Ticket Not Used
%d
",ans1);
        }
        else
        {
            print(a);
            int x=b;
            while(x!=ee)
            {
                printf(" %d",x+1);
                x=e[p[x]].u;
            }
            printf(" %d
",ee+1);
            printf("%d
%d
",a+1,ans1);
        }
    }
    return 0;
}