HNACM(八)C-最少换乘

传送门

思路是如果两个站点在同一条线路上,那么他们之间的距离置1,如果不在同一条线路上,那么他们的距离至MAX(允许的最大整数),这样再运行迪杰斯特拉算法,那么找出来的路径就是最少换乘的路径,因为换乘一次,距离便会加1,这样输出的一定是最少换乘的。

#include <bits/stdc++.h> 
#define N 510
#define ll long long
#define MAX 11111
using namespace std;
int n, m, Map[N][N], tmp[N], dis[N];
bool vis[N];
char s[10000];
int dij(){ //迪杰斯特拉算法求最少换乘 
    int i, j;
    for (i = 1; i <= n; i++){
        vis[i] = false;
        dis[i] = Map[1][i];
    }
    if (dis[n] == 1){
        dis[n];
    }
    vis[1] = true;
    for (i = 1; i < n; i++){
        int mindist = MAX;
        int u = -1;
        for (j = 1; j <= n; j++){
            if (!vis[j] && mindist > dis[j]){
                u = j;
                mindist = dis[j];
            }
        }
        if (u == n){
            return dis[n];
        }
        if (u == -1){
            return -1;
        }
        vis[u] = true;
        for (j = 1; j <= n; j++){
            if (!vis[j] && dis[j] > dis[u]+Map[u][j]){
                dis[j] = dis[u]+Map[u][j];
            }
        }
    }
}
int main(){
#ifndef ONLINE_JUDGE
// freopen("1.txt", "r", stdin);
#endif 
    int i, j, k, T, t, len, t1;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &m, &n);
        gets(s);
        for (i = 0; i <= n; i++){
            for (j = 0; j <= n; j++){
                Map[i][j] = MAX;
            }
            Map[i][i] = 0;
        }
        for (i = 0; i < m; i++){
            gets(s);//每次以字符串的方式入读 
            len = strlen(s);
            t = 0;
            k = 0;
            for (j = 0; j < len; j++){//解析字符串 
                if (s[j] >= '0' && s[j] <= '9'){
                    t *= 10;
                    t += s[j]-'0';
                }else{
                    if (t){
                        tmp[k++] = t;
                        t = 0;
                    }
                }
            }
            tmp[k] = t;
            len = k;
            for (j = 0; j <= len; j++){
                for (k = j+1; k <= len; k++){
                    Map[tmp[j]][tmp[k]] = 1;//把可以一次到达的两站置为1 
                }
            }
        }
        int ans = dij();
        if (ans > 0){
            ans--;
            printf("%d\n", ans);
        }else{
            puts("NO");//小于0说明不能到达 
        }

    }
    return 0;
}

你可能感兴趣的:(HNACM(八)C-最少换乘)