【最短路】【STL】CSU 1808 地鐵 (2016湖南省第十二屆大學生電腦程式設計競賽)

標籤：

題目連結：

　　http://acm.csu.edu.cn/OnlineJudge/problem.php?id=1808

題目大意：

　　N個點M條無向邊(N,M<=105)，每條邊屬於某一條地鐵Ci(Ci<=109),每條邊有一個耗時，如果乘Ci號線地鐵到達一個節點轉乘Cj號線地鐵離開，還需要花費|Ci-Cj|時間。

　　求1到n的最小花費時間。

題目思路：

　　【最短路】【STL】

　　d[u][Ci]表示從1到u，最後一條地鐵是Ci號線的最小耗時。按照邊做，每條邊枚舉上一個是從哪一條地鐵坐過來的，更新答案。最終統計到達n時最後是哪一號線地鐵。

　　由於Ci很大，需要開STL的set存下到每個點可能的地鐵號線，map存d[u][Ci]。

 

  1 //  2 //by coolxxx  3 //#include<bits/stdc++.h>  4 #include<iostream>  5 #include<algorithm>  6 #include<string>  7 #include<iomanip>  8 #include<map>  9 #include<stack> 10 #include<queue> 11 #include<set> 12 #include<bitset> 13 #include<memory.h> 14 #include<time.h> 15 #include<stdio.h> 16 #include<stdlib.h> 17 #include<string.h> 18 //#include<stdbool.h> 19 #include<math.h> 20 #define min(a,b) ((a)<(b)?(a):(b)) 21 #define max(a,b) ((a)>(b)?(a):(b)) 22 #define abs(a) ((a)>0?(a):(-(a))) 23 #define lowbit(a) (a&(-a)) 24 #define sqr(a) ((a)*(a)) 25 #define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b)) 26 #define mem(a,b) memset(a,b,sizeof(a)) 27 #define eps (1e-8) 28 #define J 10000 29 #define mod 1000000007  30 #define MAX 0x7f7f7f7f 31 #define PI 3.14159265358979323 32 #define N 100004 33 #define M 200004 34 using namespace std; 35 typedef long long LL; 36 int cas,cass; 37 int n,m,lll,ans; 38 int last[N],q[N]; 39 LL aans; 40 bool u[N]; 41 set<int>c[N]; 42 map<int,LL>d[N]; 43 struct xxx 44 { 45     int next,to,dd,co; 46 }a[M]; 47 void add(int x,int y,int c,int z) 48 { 49     a[++lll].to=y; 50     a[lll].dd=z; 51     a[lll].co=c; 52     a[lll].next=last[x]; 53     last[x]=lll; 54 } 55 void spfa() 56 { 57     int i,j,k,now,to,l=0,r=1; 58     set<int>::iterator ii; 59     q[1]=1; 60     for(ii=c[1].begin();ii!=c[1].end();ii++)d[1][(*ii)]=0; 61     while(l!=r) 62     { 63         now=q[l=(l+1)%N]; 64         if(now==n)continue; 65         u[now]=0; 66         for(i=last[now];i;i=a[i].next) 67         { 68             to=a[i].to; 69             if(d[to].find(a[i].co)==d[to].end()) 70                 d[to][a[i].co]=MAX; 71             for(ii=c[now].begin();ii!=c[now].end();ii++) 72             { 73                 if(d[now].find((*ii))==d[now].end())continue; 74                 if(d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co)>aans)continue; 75                 if(d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co)<d[to][a[i].co]) 76                 { 77                     d[to][a[i].co]=d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co); 78                     if(to==n) 79                     { 80                         aans=min(aans,d[to][a[i].co]); 81                         continue; 82                     } 83                     if(!u[to]) 84                     { 85                         u[to]=1; 86                         q[r=(r+1)%N]=to; 87                     } 88                 } 89             } 90         } 91     } 92 } 93 int main() 94 { 95     #ifndef ONLINE_JUDGE 96 //    freopen("1.txt","r",stdin); 97 //    freopen("2.txt","w",stdout); 98     #endif 99     int i,j,k;100     int x,y,z;101 //    for(scanf("%d",&cass);cass;cass--)102 //    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)103 //    while(~scanf("%s",s+1))104     while(~scanf("%d",&n))105     {106         for(i=1;i<=n;i++)107         {108             c[i].clear();109             d[i].clear();110         }111         mem(last,0);mem(u,0);lll=0;112         scanf("%d",&m);113         for(i=1;i<=m;i++)114         {115             scanf("%d%d%d%d",&x,&y,&j,&z);116             add(x,y,j,z);117             add(y,x,j,z);118             c[x].insert(j);c[y].insert(j);119         }120         aans=100000000000000;121         spfa();122         printf("%lld\n",aans);123     }124     return 0;125 }126 /*127 //128 129 //130 */

View Code

 

【最短路】【STL】CSU 1808 地鐵 (2016湖南省第十二屆大學生電腦程式設計競賽)