CSUOJ 1808 地铁（用map储存DP数组的DP）

1808: 地铁

Time Limit: 5 Sec   Memory Limit: 128 MB
Submit: 206   Solved: 22
[ Submit][ Status][ Web Board]

Description

 Bobo 居住在大城市 ICPCCamp。

ICPCCamp 有 n 个地铁站，用 1,2,…,n 编号。 m 段双向的地铁线路连接 n 个地铁站，其中第 i 段地铁属于 c _i 号线，位于站 a _i,b _i 之间，往返均需要花费 t _i 分钟（即从 a _i 到 b _i 需要 t _i 分钟，从 b _i 到 a _i 也需要 t _i 分钟）。

众所周知，换乘线路很麻烦。如果乘坐第 i 段地铁来到地铁站 s，又乘坐第 j 段地铁离开地铁站 s，那么需要额外花费 |c _i-c _j | 分钟。注意，换乘只能在地铁站内进行。

Bobo 想知道从地铁站 1 到地铁站 n 所需要花费的最小时间。

Input

输入包含不超过 20 组数据。

每组数据的第一行包含两个整数 n,m (2≤n≤10 ⁵,1≤m≤10 ⁵).

接下来 m 行的第 i 行包含四个整数 a _i,b _i,c _i,t _i (1≤a _i,b _i,c _i≤n,1≤t _i≤10 ⁹).

保证存在从地铁站 1 到 n 的地铁线路（不一定直达）。

Output

对于每组数据，输出一个整数表示要求的值。

Sample Input

3 3
1 2 1 1
2 3 2 1
1 3 1 1
3 3
1 2 1 1
2 3 2 1
1 3 1 10
3 2
1 2 1 1
2 3 1 1

Sample Output

1
3
2

HINT

Source

湖南省第十二届大学生计算机程序设计竞赛

题目大意：

    在一个有N个地铁站的城市中，有M条线路，它们属于不同的地铁线。在换乘时需要花费| ci - cj |的时间。其它和最短路一样。求从站点1到站点N的最小时间。

解题思路：

    开始就只是改了一下SPFA的松弛操作结果WA了，后来在和队友讨论的时候发现必须用一个二维数组dp[i][j]记录到达第i个站点最后坐j号线的最小时间。由于M最大为1e5所以当N很大时这个图一定非常稀疏，所以枚举每个站点有效的各个状态然后按照SPFA的方式进行状态转移的大时间复杂度虽然是O(N^2)，但是由于这个边数的制约，所以时间上会比N^2小的多，不会超时。不过这样还有一个问题，如果使用二维数组存dp[i][j]初始化就会超时也不好利用这个边数的制约条件，所以我们用一个map数组来储存，这样初始化只需要把每个顶点都clear()一下，转移时用迭代器便利就可以了。

    不过非常可惜，比赛时和队友讨论出这个解法后只剩下10min了，没有来得及A掉，比赛后才A。

    叉姐一句普通的最短路就让本鶸想了4h，果然还是too young too simple了。

AC代码：

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<math.h>
#include<iterator>
#include<stack>
using namespace std;
#define LL long long
#define INF (0x3f3f3f3f3f3f3f3fll)
#define fi first
#define se second

const int maxn=100000+3;

struct Edge
{
    LL v,cost,c;
    Edge(LL v,LL cost,LL c):v(v),cost(cost),c(c){}
};

vector<Edge> G[maxn];
bool vis[maxn];
map<LL,LL> dp[maxn];//dp[i][j]表示到第i个城市最后坐地铁j的最少时间
int N,M;

void SPFA()
{
    for(int i=1;i<=N;++i)
        vis[i]=false;
    vis[1]=true;
    queue<LL> que;
    que.push(1);
    while(!que.empty())
    {
        LL u=que.front(); que.pop();
        vis[u]=false;
        for(int i=0;i<G[u].size();++i)
        {
            LL v=G[u][i].v;
            if(v==1)//不可能更新起点
                continue;
            if(u==1)//从起点出发的边特判
            {
                if(dp[v][G[u][i].c]==0)//dp[i][j]==0表示还没有使用过
                {
                    dp[v][G[u][i].c]=G[u][i].cost;
                    if(!vis[v])
                    {
                        vis[v]=true;
                        que.push(v);
                    }
                }
                else 
                {
                    if(dp[v][G[u][i].c]>G[u][i].cost)
                    {
                        dp[v][G[u][i].c]=G[u][i].cost;
                        if(!vis[v])
                        {
                            vis[v]=true;
                            que.push(v);
                        }
                    }
                }
            }
            for(map<LL,LL>::iterator it=dp[u].begin();it!=dp[u].end();++it)//枚举可以有值的各个路线
            {
                if(dp[v][G[u][i].c]==0)
                {
                    dp[v][G[u][i].c]=dp[u][(it->fi)]+G[u][i].cost+abs((it->fi)-G[u][i].c);
                    if(!vis[v])
                    {
                        vis[v]=true;
                        que.push(v);
                    }
                }
                else
                {
                    if(dp[v][G[u][i].c]>dp[u][(it->fi)]+G[u][i].cost+abs((it->fi)-G[u][i].c))
                    {
                        dp[v][G[u][i].c]=dp[u][(it->fi)]+G[u][i].cost+abs((it->fi)-G[u][i].c);
                        if(!vis[v])
                        {
                            vis[v]=true;
                            que.push(v);
                        }
                    }
                }
            }
        }
    }
}

int main()
{
    while(~scanf("%d%d",&N,&M))
    {
        for(int i=1;i<=N;++i)//初始化邻接表和dp数组
        {
            G[i].clear();
            dp[i].clear();
        }
        for(int i=0;i<M;++i)
        {
            LL a,b,c,t;
            scanf("%lld%lld%lld%lld",&a,&b,&c,&t);
            G[a].push_back(Edge(b,t,c));
            G[b].push_back(Edge(a,t,c));
        }
        SPFA();
        LL ans=INF;
        for(int i=1;i<=N;++i)
            if(dp[N][i])
                ans=min(ans,dp[N][i]);
        printf("%lld\n",ans);
    }
    
    return 0;
}