CSU 1808 地铁

Posted 2020-08-06 Fighting Heart

湖南省第十二届大学生计算机程序设计竞赛$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;
}