【最短路】【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=0,r=1;
	set<int>::iterator ii;
	q[1]=1;
	for(ii=c[1].begin();ii!=c[1].end();ii++)d[1][(*ii)]=0;
	while(l!=r)
	{
		now=q[l=(l+1)%N];
		if(now==n)continue;
		u[now]=0;
		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]=1;
						q[r=(r+1)%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=1;i<=n;i++)
		{
			c[i].clear();
			d[i].clear();
		}
		mem(last,0);mem(u,0);lll=0;
		scanf("%d",&m);
		for(i=1;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=100000000000000;
		spfa();
		printf("%lld\n",aans);
	}
	return 0;
}
/*
//

//
*/