问题 K: 最优乘车

题目描述

某    城市是一个旅游胜地，每年都有成千上万的人前来观光。为方便游客，巴士公司在各个旅游景点及宾馆、饭店等地都设置了巴士站并开通了一些单程巴士线路。每条单程巴士线路从某个巴士站出发，依次经过若干个巴士站，最终到达终点巴士站。

    一名旅客最近到该城市旅游，他很想去S公园游玩，但如果从他所在的饭店没有一路巴士可以直接到达S公园，则他可能要先乘某一路巴士坐几站，再下来换乘同一站台的另一路巴士，这样换乘几次后到达S公园。

    现在用整数1, 2,..., n给该城市所有的巴士站编号，约定这名旅客所在饭店的巴士站编号为 1，S公园巴士站的编号为n。

    写一个程序，帮助这名旅客寻找一个最优乘车方案，是他在从饭店乘车到S公园的过程中换车的次数最少。

输入

    第1行有两个整数m和n（1 <= m <= 100， 1 < N <= 500），表示开通了m条单程巴士线路，总共有n个车站。

    从第2行到第m+1行依次给出了每条巴士线路的信息。其中第i+1行给出的是第i条巴士线路的信息，从左至右按运行顺序依次给出了该线路上的所有站号，相邻两个站号之间用一个空格隔开。

#include <bits/stdc++.h>
using namespace std;
const int N = 510, INF = 0x3f3f3f3f;
int dis[N];  // 存储a号点到每个点的最短距离
int path[N][N];//边的权值
int station[N];
bool st[N];// 存储每个点的最短距离是否已经确定
vector<int> v;//存放每条路线的所有站点
int n, m;
void dijkstra() {
    //int可以直接0x3f，double的话0x7f
    memset(dis, INF, sizeof dis);
    dis[1] = 0;
    //n-1次迭代
    for (int i = 0; i < n - 1; i++) {
        int t = -1;     // 在还未确定最大路的点中，寻找距离最小的点
        for (int j = 1; j <= n; ++j) {
            if (!st[j] && (t == -1 || dis[t] > dis[j]))
                t = j;
        }
        st[t] = true;//起点到t的最短路已确定
        //用t更新到其他n个点的距离
        for (int j = 1; j <= n; ++j) {
            if (dis[j] > dis[t] + path[t][j])
                dis[j] = dis[t] + path[t][j];
        }
    }
}
void generate(string s) {
    s.append(" ");//最后一个数的结束条件
    v.clear();
    int num = 0;
    for (int i = 0; i < s.size(); ++i) {
        if (s[i] == ' ') {
            v.push_back(num);
            num = 0;
        }
        else {
            num = num * 10 + (s[i] - '0');
        }
    }
}
 
int main() {
    cin.tie(0);
    memset(path, INF, sizeof path);
    cin >> m >> n;
    string s;
    getline(cin, s);//吸收换行符
    while (m--) {
        getline(cin, s);
        stringstream ssin(s);
        int cnt = 0, k;
        while (ssin >> k) station[cnt++] = k;
        for (int i = 0; i < cnt; i++)
            for (int j = i + 1; j < cnt; j++)
                path[station[i]][station[j]] = 1;          // 从 station[i] 到 station[j] 只要坐一次车
    }
    dijkstra();
    if (dis[n] == INF)
        cout << "NO";
    else
        cout << dis[n] - 1;
    return 0;
}