DIJ最短路 - Subway POJ - 2502

最短路 - Subway POJ - 2502

题意：

小k要从家去学校，他可以选择步行或者地铁，步行的速度是10km/h，地铁的速度是40km/h。假设小k非常的幸运，每次他到地铁站就立刻会有地铁。小k可以随意上下地铁，并且可以在地铁线路之间转换。所有的地铁运行都是双向的。

Input

输入第一行包括家和学校的x，y坐标，接下来是若干条地铁线。
每条地铁线包括站点的x，y坐标，有序排列，假设地铁在相邻的两站之间直线运行，每条地铁线至少有两个站，地铁线的描述以-1，-1结束。
该市最多有200个地铁站。

Output

输出是上学所需的时间，四舍五入到最近的分钟，采取最快的路线。

Sample Input

0 0 10000 1000
0 200 5000 200 7000 200 -1 -1 
2000 600 5000 600 10000 600 -1 -1

Sample Output

21

分析：

①、任意两点a和b之间都可步行连通，权值为：$\frac{dis(a,b)}{10km/h}\times 60(min)$

②、同一条地铁线上，相邻两站之间可以乘坐地铁通行，权值为：$\frac{dis(a,b)}{40km/h}\times 60(min)$

③、特别注意： 同一条地铁线上，不相邻的两点，不能够直接用距离除车速，因为这两点是不能够直达的。

④、所有点是需要保存下来的，因为我们建图的时候需要。每个点都有一个编号，表示该点所在的地铁线编号，便于建图。

时间复杂度分析： 至多200个点，约200²条边，采用dijkstra算法，时间复杂度$O(nlo{g}_{2}m)$

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<cmath>

#define x first
#define y second

using namespace std;

const int N = 210, M = 41000;
const double v1 = 10000.0, v2 = 40000.0;
typedef pair<int, int> PII;
typedef pair<double, int> PDI;

int e[M], ne[M], h[N], idx;
double w[M];
PII v[N];
int n, m;
int station[N], sub_cnt = 1;
double dis[N];
bool st[N];

double get_dis(PII a, PII b)
{
    double dx = a.x - b.x, dy = a.y - b.y;
    return sqrt(dx * dx + dy * dy);
}

void add(int a, int b, double c)
{
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

void add(double V, int a, int b)
{
    add(a, b, get_dis(v[a], v[b]) / V * 60), add(b, a, get_dis(v[a], v[b]) / V * 60);
}

void build()
{
    memset(h, -1, sizeof h);
    for(int i=2;i<n-1;i++)
        if(station[i] == station[i + 1])
            add(v2, i, i+1);

    for(int i=0;i<n;i++)
        for(int j=i+1;j<n;j++)
            add(v1, i, j);
}

double dijkstra()
{
    for(int i=0;i<N;i++) dis[i] = 1e20;
    priority_queue<PDI, vector<PDI>, greater<PDI> > heap;
    dis[0] = 0;
    heap.push({0, 0});

    while(heap.size())
    {
        PDI u = heap.top();
        heap.pop();
        
        int id = u.y;
        if(st[id]) continue;
        st[id] = true;
        
        for(int i=h[id];~i;i=ne[i])
        {
            int j = e[i];
            if(dis[j] > dis[id] + w[i])
            {
                dis[j] = dis[id] + w[i];
                heap.push({dis[j], j});
            }
        }
    }
    
    return dis[1];
}

int main()
{
    for(int i=0;i<2;i++) scanf("%d%d", &v[i].x, &v[i].y);
    n = 2;

    PII t;
    while(~scanf("%d%d", &t.x, &t.y))
    {
        if(t == make_pair(-1, -1)) 
        {
            sub_cnt ++;
            continue;
        }
        station[n] = sub_cnt;
        v[n ++] = t;
    }

    build();
    printf("%.0f\n", dijkstra());
    
    return 0; 
}