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

时间：2023-03-08 17:35:35

题目链接：

　　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]。

 //

 //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 ;

 }

 /*

 //

 //

 */

相关文章