【最短路】【STL】CSU 1808 地铁 (2016湖南省第十二届大学生计算机程序设计竞赛)

题目链接：

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

题目大意：

　　N个点M条无向边(N,M<=10⁵)，每条边属于某一条地铁Ci(Ci<=10⁹),每条边有一个耗时，如果乘Ci号线地铁到达一个节点换乘Cj号线地铁离开，还需要花费|Ci-Cj|时间。

　　求1到n的最小花费时间。

题目思路：

　　【最短路】【STL】

　　d[u][Ci]表示从1到u，最后一条地铁是Ci号线的最小耗时。按照边做，每条边枚举上一个是从哪一条地铁坐过来的，更新答案。最终统计到达n时最后是哪一号线地铁。

　　由于Ci很大，需要开STL的set存下到每个点可能的地铁号线，map存d[u][Ci]。

 //
 //by coolxxx
 //#include<bits/stdc++.h>
 #include<iostream>
 #include<algorithm>
 #include<string>
 #include<iomanip>
 #include<map>
 #include<stack>
 #include<queue>
 #include<set>
 #include<bitset>
 #include<memory.h>
 #include<time.h>
 #include<stdio.h>
 #include<stdlib.h>
 #include<string.h>
 //#include<stdbool.h>
 #include<math.h>
 #define min(a,b) ((a)<(b)?(a):(b))
 #define max(a,b) ((a)>(b)?(a):(b))
 #define abs(a) ((a)>0?(a):(-(a)))
 #define lowbit(a) (a&(-a))
 #define sqr(a) ((a)*(a))
 #define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
 #define mem(a,b) memset(a,b,sizeof(a))
 #define eps (1e-8)
 #define J 10000
 #define mod 1000000007
 #define MAX 0x7f7f7f7f
 #define PI 3.14159265358979323
 #define N 100004
 #define M 200004
 using namespace std;
 typedef long long LL;
 int cas,cass;
 int n,m,lll,ans;
 int last[N],q[N];
 LL aans;
 bool u[N];
 set<int>c[N];
 map<int,LL>d[N];
 struct xxx
 {
     int next,to,dd,co;
 }a[M];
 void add(int x,int y,int c,int z)
 {
     a[++lll].to=y;
     a[lll].dd=z;
     a[lll].co=c;
     a[lll].next=last[x];
     last[x]=lll;
 }
 void spfa()
 {
     int i,j,k,now,to,l=,r=;
     set<int>::iterator ii;
     q[]=;
     for(ii=c[].begin();ii!=c[].end();ii++)d[][(*ii)]=;
     while(l!=r)
     {
         now=q[l=(l+)%N];
         if(now==n)continue;
         u[now]=;
         for(i=last[now];i;i=a[i].next)
         {
             to=a[i].to;
             if(d[to].find(a[i].co)==d[to].end())
                 d[to][a[i].co]=MAX;
             for(ii=c[now].begin();ii!=c[now].end();ii++)
             {
                 if(d[now].find((*ii))==d[now].end())continue;
                 if(d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co)>aans)continue;
                 if(d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co)<d[to][a[i].co])
                 {
                     d[to][a[i].co]=d[now][(*ii)]+a[i].dd+abs((*ii)-a[i].co);
                     if(to==n)
                     {
                         aans=min(aans,d[to][a[i].co]);
                         continue;
                     }
                     if(!u[to])
                     {
                         u[to]=;
                         q[r=(r+)%N]=to;
                     }
                 }
             }
         }
     }
 }
 int main()
 {
     #ifndef ONLINE_JUDGE
 //    freopen("1.txt","r",stdin);
 //    freopen("2.txt","w",stdout);
     #endif
     int i,j,k;
     int x,y,z;
 //    for(scanf("%d",&cass);cass;cass--)
 //    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
 //    while(~scanf("%s",s+1))
     while(~scanf("%d",&n))
     {
         for(i=;i<=n;i++)
         {
             c[i].clear();
             d[i].clear();
         }
         mem(last,);mem(u,);lll=;
         scanf("%d",&m);
         for(i=;i<=m;i++)
         {
             scanf("%d%d%d%d",&x,&y,&j,&z);
             add(x,y,j,z);
             add(y,x,j,z);
             c[x].insert(j);c[y].insert(j);
         }
         aans=;
         spfa();
         printf("%lld\n",aans);
     }
     return ;
 }
 /*
 //

 //
 */