湖南省第十二届大学生计算机程序设计竞赛$F$题。
最短路。
如果只记录到某个节点的最短路,显然是错误的。这题的状态有两个量决定,即到了哪一个节点,最后一辆乘坐的是几号线。这个可以用$map$记录一下。
要注意的是:如果用$SPFA$,因为要标记某个状态是否在队列中,还需要额外开一个$map$,这样可能导致超时;用$dijkstra$$+$优先队列优化的话就可以通过。
#pragma comment(linker, "/STACK:1024000000,1024000000") #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #include<vector> #include<map> #include<set> #include<queue> #include<stack> #include<iostream> using namespace std; typedef long long LL; const double pi=acos(-1.0),eps=1e-6; void File() { freopen("D:\\in.txt","r",stdin); freopen("D:\\out.txt","w",stdout); } template <class T> inline void read(T &x) { char c = getchar(); x = 0;while(!isdigit(c)) c = getchar(); while(isdigit(c)) { x = x * 10 + c - ‘0‘; c = getchar(); } } const LL INF=(LL)1e17; const int maxn=200010; struct Edge { int u,v,c; LL t; int nx; }e[maxn]; int h[maxn],sz; int n,m; LL ans[maxn]; void add(int u,int v,int c,LL t) { e[sz].u=u; e[sz].v=v; e[sz].c=c; e[sz].t=t; e[sz].nx=h[u]; h[u]=sz++; } struct Node { int p,id; LL dis; Node(int P,int ID,LL DIS) { dis=DIS, p=P, id=ID; } bool operator < (const Node &a) const { if(dis==a.dis&&p==a.p) return id>a.id; if(dis==a.dis) return p>a.p; return dis>a.dis; } }; struct X { int p,id; X (int P,int ID) { p=P, id=ID; } bool operator < (const X &a) const { if(p==a.p) return id>a.id; return p>a.p; } }; LL ABS(LL a) { if(a>=0) return a; return -a; } void dij() { map<X,LL>f; priority_queue<Node>Q; Q.push(Node(1,0,0)); for(int i=1;i<=n;i++) ans[i]=INF; ans[1]=0; while(!Q.empty()) { Node top=Q.top(); Q.pop(); ans[top.p]=min(ans[top.p],top.dis); if(top.dis>f[X(top.p,top.id)]) continue; for(int i=h[top.p];i!=-1;i=e[i].nx) { LL COST; if(top.p==1) COST=0; else COST=ABS((LL)e[i].c-(LL)top.id); LL pp=f[X(e[i].v,e[i].c)]; if((pp==0&&e[i].v!=1)||top.dis+COST+e[i].t<pp) { f[X(e[i].v,e[i].c)]=top.dis+COST+e[i].t; Q.push(Node(e[i].v,e[i].c,top.dis+COST+e[i].t)); } } } } int main() { while(~scanf("%d%d",&n,&m)) { memset(h,-1,sizeof h); sz=0; for(int i=1;i<=m;i++) { int u,v,c; LL t; scanf("%d%d%d%lld",&u,&v,&c,&t); add(u,v,c,t); add(v,u,c,t); } dij(); cout<<ans[n]<<endl; } return 0; }
时间: 2024-09-30 07:13:20