hdu 3499 Flight

分层图最短路模板题。

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf = 1e18;
const int M = 1e5+7;
int n,m,cnt,head[M],flag[M][2];
struct edge
{
    int v,nex;ll w;
}e[M*5];
void init(){
    cnt=0;memset(head,-1,sizeof(head));memset(flag,0,sizeof(flag));
}
void add(int u,int v,ll w){
    e[++cnt].v=v;e[cnt].w=w;e[cnt].nex=head[u];
    head[u]=cnt;
}
struct node
{
    int index,level;ll dis;
    bool operator < (const node &k)const{
        return dis>k.dis;
    }
}d[M][2];
void dijkstra(int s){
    for(int i=1;i<=n;i++)
        for(int j=0;j<=1;j++) d[i][j].dis=inf,d[i][j].index=i,d[i][j].level=j;
    priority_queue<node> q;d[s][0].dis=0;q.push(d[s][0]);
    while(!q.empty()){
        int u=q.top().index,lv=q.top().level;q.pop();
        //cout<<u<<" "<<lv<<" "<<d[u][lv].dis<<endl;
        if(flag[u][lv]) continue;
        flag[u][lv]=1;

        for(int i=head[u];~i;i=e[i].nex){
            int v=e[i].v;ll w=e[i].w;
            if(d[v][lv].dis>d[u][lv].dis+w){
                d[v][lv].dis=d[u][lv].dis+w;
                q.push(d[v][lv]);
            }
            if(lv==0&&d[v][lv+1].dis>d[u][lv].dis+w/2){
                d[v][lv+1].dis=d[u][lv].dis+w/2;
                q.push(d[v][lv+1]);
            }
        }

    }
    
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        map<string,int> mp;
        init();int tmp=0;
        string st,ed;ll w;
        for(int i=1;i<=m;i++){
            cin>>st>>ed>>w;
            if(!mp[st]) mp[st]=++tmp;
            if(!mp[ed]) mp[ed]=++tmp;
            add(mp[st],mp[ed],w);
            //cout<<mp[st]<<" "<<mp[ed]<<endl;
        }
        cin>>st>>ed;
        if(m==0){
            printf("-1\n");
            continue;
        }
        //cout<<mp[st]<<" "<<mp[ed]<<endl;
        dijkstra(mp[st]);
        ll ans=min(d[mp[ed]][0].dis,d[mp[ed]][1].dis);
        ans==inf?printf("-1\n"):printf("%lld\n",ans);
    }
    return 0;
}

View Code

 

转载于:https://www.cnblogs.com/LMissher/p/9573722.html