hdu 3499 最短路变形

// hdu 3499.cpp : 定义控制台应用程序的入口点。
/*
题意描述：给定一个带权值的有向图，求给定2点之间的最短距离。但是你有一次机会将某条边的权值变为原来的一半。
解题思路: 1,在正向图中求一次dijk，求起点到任何点的距离dis_start[i]
    2,在反向图中求一次dijk, 求终点到任何点的距离dis_end[i]
    3,枚举所有的边
     ans = min{dis_start[i] + weight[i][j] / 2 + dis_end[j]}
    题目数据很BT，所以要用到邻接表，还有dis要用到__int64
*/

#include "stdafx.h" #include <iostream> #include <map> #include <string> #include <algorithm> #include <set> using namespace std; const int MAXN = 100000+10; const int MAXM = 500000+10; const __int64 INF = (_I64_MAX)/2; typedef struct { int v; int w; int next; }Edge; //下标为0 均为正向图 Edge e[2][MAXM]; //e[0] 保存正向图 e[1]保存反向图 int pre[2][MAXM]; int from[MAXM],to[MAXM],weight[MAXM]; //保存原来的边 int visit[MAXN]; int index[2]; __int64 dis[2][MAXN]; int sign; map<string,int>M; void init() { index[0] = index[1] = 0; memset(pre,-1,sizeof(pre)); } void insert(int u, int v, int w, int t) { e[t][index[t]].v = v; e[t][index[t]].w = w; e[t][index[t]].next = pre[t][u]; pre[t][u] = index[t]++; } struct cmp { bool operator() (const int &a, const int &b) const { if(dis[sign][a] == dis[sign][b]) { return a < b; } return dis[sign][a] < dis[sign][b]; } }; void dijkstra(int st) { set<int,cmp>S; S.clear(); memset(visit,0,sizeof(visit)); //memset(dis[sign],0x7f,sizeof(dis[sign])); for(int i=0; i<MAXN; i++) { dis[sign][i] = INF; } dis[sign][st] = 0; S.insert(st); while(!S.empty()) { st = *S.begin(); S.erase(st); visit[st] = true; for(int i=pre[sign][st]; i!=-1; i=e[sign][i].next) { if(!visit[e[sign][i].v] && dis[sign][e[sign][i].v] > dis[sign][st] + e[sign][i].w) { S.erase(e[sign][i].v); //必须先删除该点，否则下面成功插入 dis[sign][e[sign][i].v] = dis[sign][st] + e[sign][i].w; S.insert(e[sign][i].v); } } } } int main() { int n,m; int countN; int num_s,num_t; string s,t; int price; while(cin>>n>>m) { M.clear(); countN = 0; init(); for(int i=0; i<m; i++) { cin>>s>>t>>price; if(M.find(s) == M.end()) { M[s] = countN++; } if(M.find(t) == M.end()) { M[t] = countN++; } num_s = M[s],num_t = M[t]; //正向图 insert(num_s,num_t,price,0); //反向图 insert(num_t,num_s,price,1); //将所有的边先保存起来 from[i] = num_s; to[i] = num_t; weight[i] = price; } cin>>s>>t; if(M.find(s) == M.end() || M.find(t) == M.end()) //不存在该点 { cout<<"-1"<<endl; continue; } num_s = M[s]; num_t = M[t]; if(num_s == num_t) //原点与起点同点 { cout<<"0"<<endl; continue; } //正向图,起点到任何点的最短距离 sign = 0; dijkstra(num_s); if(dis[0][num_t] == INF) //不可达 { cout<<"-1"<<endl; continue; } //反向图，终点到任何点的最短距离 sign = 1; dijkstra(num_t); __int64 ans = INF; for(int i=0; i<m; i++) { if(dis[0][from[i]] + dis[1][to[i]] + weight[i] / 2 < ans) { ans = dis[0][from[i]] + dis[1][to[i]] + weight[i] / 2; } } printf("%I64d/n",ans); } return 0; }