csu 1307 City Tour dijkstra+并查集 最短路 解题报告

Description

Alice想要从城市A出发到城市B,由于Alice最近比较穷(不像集训队陈兴老师是个rich second),所以只能选择做火车从A到B。不过Alice很讨厌坐火车,火车上人比较多,比较拥挤,所以Alice有很严格的要求:火车的相邻两站间的最大距离尽可能的短,这样Alice就可以在停站的时候下车休息一下。当然Alice希望整个旅途比较短。

Input

有多组测试数据。
每组测试数据的第一行有两个整数N,M,A,B(N<=2000, M<=50000, N >=2, A,B<=N),其中N是城市的个数,M是城市间通火车的个数。
A,B是Alice起始的城市与目的地城市,城市的标号从1开始。
接下来的M行每行三个整数u,v,w表示从u到v和从v到u有一条铁路,距离为w, u,v<=N, w<=10000。

Output

对于每组测试数据输出满足Alice要求的从A到B的最短距离。

Sample Input

3 3 1 2
1 2 80
1 3 40
2 3 50
3 3 1 2
1 2 90
1 3 10
2 3 20
4 5 1 4
1 2 8
1 4 9
1 3 10
2 4 7
3 4 8

Sample Output

90
30
15

思路:

首先按边从小到大排序,类似kruskal建树,直到s与t点在同一个联通分量,范围当前边的最大值,因为已经排序了,所以当前边即是最大。

上一步求得了s到t相邻最小的最大,然后用dijkstra求小于这条边的最短路,即可。

代码:

#include 
#include 
#include 
#include 
#define MAXN 2020
#define MAXM 55000
#define INF 0x3f3f3f3f
using namespace std;

struct node
{
    int x, y, v;
}w[MAXM * 2];

int f[MAXN], n, m, s, t, maxE;
int dis[MAXN], map[MAXN][MAXN], vis[MAXN];

int cmp(node a, node b)
{
    return a.v < b.v;
}

int find(int x)
{
    return f[x] == x ? x : (f[x] = find(f[x]));
}

int cal()
{
    int me = 0;
    for(int i = 1; i <= n; i++) f[i] = i;
    sort(w, w + m, cmp);
    for(int i = 0; i < m; i++)
    {
        int x = find(w[i].x);
        int y = find(w[i].y);
        if(x != y)
        {
            f[x] = y;
            me = max(me, w[i].v);
            if(find(s) == find(t))
            {
                return me;
            }
        }
    }
    return me;
}

int dijkstra()
{
    memset(vis, 0, sizeof(vis));
    memset(dis, 0x3f, sizeof(dis));
    for(int i = 1; i <= n; i++)
    {
        if(map[s][i] > maxE) continue;
        dis[i] = map[s][i];
    }
    vis[s] = 1;
    dis[s] = 0;
    while(true)
    {
        int k = -1, mixv = INF;
        for(int i = 1; i <= n; i++)
        {
            if(!vis[i] && mixv > dis[i])
            {
                k = i;
                mixv = dis[i];
            }
        }
        if(k == -1) break;
        vis[k] = 1;
        for(int i = 1; i <= n; i++)
        {
            if(!vis[i] && (dis[i] > dis[k] + map[k][i]) && map[k][i] <= maxE)
            {
                dis[i] = dis[k] + map[k][i];
            }
        }
    }
    /*for(int i = 1; i <= n; i++)
    {
        printf("%d ", dis[i]);
    }
    printf("\n");*/
    return dis[t];
}

int main()
{
    while(scanf("%d%d%d%d", &n, &m, &s, &t) != EOF)
    {
        memset(map, 0x3f, sizeof(map));
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d%d", &w[i].x, &w[i].y, &w[i].v);
            if(w[i].v < map[w[i].x][w[i].y])
                map[w[i].x][w[i].y] = map[w[i].y][w[i].x] = w[i].v;
        }
        maxE = cal();
        //printf("%d\n", maxE);
        printf("%d\n", dijkstra());
    }
    return 0;
}


你可能感兴趣的:(最短路,图论)