hdu3499(分层图最短路)

链接：点击打开链接

题意：给出n个城市和m条边，和走每条边的费用，又一次免费走一条边的机会，问从一个城市走到另一个城市最小的费用是多少

代码：

#include <map>
#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
const long long INF=0x3f3f3f3f3f3f3f3fll;
using namespace std;
struct node{
    int to,cost;
};
int V;
long long d[200005];
vector<node> G[200005];
typedef pair<int,int> P;
void dijkstra(int s){
    int i,v;
	priority_queue<P,vector<P>,greater<P> >que;
	memset(d,INF,sizeof(d));
	d[s]=0;
	que.push(P(0,s));
	while(!que.empty()){
		P p=que.top(); que.pop();
		v=p.second;
		if(d[v]<p.first)
        continue;
		for(i=0;i<G[v].size();i++){
			node e=G[v][i];
			if(d[e.to]>d[v]+e.cost){
				d[e.to]=d[v]+e.cost;
				que.push(P(d[e.to],e.to));
			}
		}
	}
}                                               //最短路穆模板
int main(){
    int m,i,j,id,st,en,num;
    long long ans;
    map<string,int> mp;
    string s1,s2;
    while(scanf("%d%d",&V,&m)!=EOF){
        id=1;
        mp.clear();
        for(i=0;i<=2*V;i++)
        G[i].clear();
        for(i=0;i<m;i++){
        cin>>s1>>s2>>num;
        if(!mp[s1])                             //用map统计字符串的序号
        mp[s1]=id++;                            //也可以用tire树 
        if(!mp[s2])
        mp[s2]=id++;
        G[mp[s1]].push_back((node){mp[s2],num});
        G[mp[s1]+V].push_back((node){mp[s2]+V,num});
        G[mp[s1]].push_back((node){mp[s2]+V,num/2});
        }                                       //因为只有一条免费的路，因此建两层图
        cin>>s1>>s2;                            //从上层跑到下层就相当于用了一次免费的机会
        if(!mp[s1])                             //注意起点和终点可能是没出现过的串
        mp[s1]=id++;
        if(!mp[s2])
        mp[s2]=id++;
        st=mp[s1],en=mp[s2];
        dijkstra(st);
        ans=min(d[en],d[en+V]);
        if(ans==INF)
        ans=-1;
        printf("%I64d\n",ans);
    }
    return 0;
}