CSU 1808 地铁(dijkstra+heap)

ICPCCamp 有 n 个地铁站，用 1,2,…,n 编号。 m 段双向的地铁线路连接 n 个地铁站，其中第 i 段地铁属于 ci 号线，位于站 ai,bi 之间，往返均需要花费 ti 分钟（即从 ai 到 bi 需要 ti 分钟，从 bi 到 ai 也需要 ti 分钟）。
众所周知，换乘线路很麻烦。如果乘坐第 i 段地铁来到地铁站 s，又乘坐第 j 段地铁离开地铁站 s，那么需要额外花费 |ci-cj | 分钟。注意，换乘只能在地铁站内进行。
Bobo 想知道从地铁站 1 到地铁站 n 所需要花费的最小时间。

---

$题意很简单，就是和以前的最短路有些不同$
$这题如果把点当作状态的话有些不妥$
$其实这题因为有了额外花费这个条件，应该把边当作状态$
$dis[i]表示走过i这条边打到edge[i].v的最少花费$
$这样的话就2*m条边$
$把边看作点来做最短路，但是这题的边数未知$
$所以spfa会TLE，所以要用dijkstra+heap$

---

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           100005
#define   MAXN          1000005
#define   maxnode       205
#define   sigma_size    2
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
//const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-9;
const LL     mod   = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

struct Edge{
    int v,next,num,c;
}edge[MAX*2];

struct Node{
    int id,val;
    bool operator<(const Node &a)const{
        return val>a.val;
    }
}x;

int head[MAX];
LL dis[MAX*2];
int vis[MAX*2];
int tot;

void add_edge(int a,int b,int c,int d){
    edge[tot]=(Edge){b,head[a],c,d};
    head[a]=tot++;
    edge[tot]=(Edge){a,head[b],c,d};
    head[b]=tot++;
}

LL dijkstra(int s,int t){
    priority_queue<Node> q;
    for(int i=0;i<tot;i++){
        dis[i]=1e18;
        vis[i]=0;
    }
    for(int i=head[s];i!=-1;i=edge[i].next){
        x=(Node){i,edge[i].c};
        dis[i]=edge[i].c;
        q.push(x);
    }
    LL ans=1e18;
    while(!q.empty()){
        x=q.top();
        q.pop();
        int p=x.id;
        if(vis[p]) continue;
        vis[p]=1;
        int u=edge[p].v;
        if(u==t) ans=min(ans,dis[p]);
        for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].v;
            if(!vis[i]&&dis[i]>dis[p]+edge[i].c+abs(edge[i].num-edge[p].num)){
                dis[i]=dis[p]+edge[i].c+abs(edge[i].num-edge[p].num);
                q.push((Node){i,dis[i]});
            }
        }
    }
    return ans;
}
int main() {
    int n,m;
    while(cin>>n>>m){
        tot=0;
        for(int i=1;i<=n;i++) head[i]=-1;
        for(int i=0;i<m;i++){
            int a,b,c,d;
            RI(a);RI(b);RI(c);RI(d);
            add_edge(a,b,c,d);
        }
        cout<<dijkstra(1,n)<<endl;
    }
    return 0;
}