森森喜欢坐地铁。这个假期,他终于来到了传说中的地铁之城——魔都,打算好好过一把坐地铁的瘾!
魔都地铁的计价规则是:起步价 2 元,出发站与到达站的最短距离(即计费距离)每 K 公里增加 1 元车费。
例如取 K = 10,动安寺站离魔都绿桥站为 40 公里,则车费为 2 + 4 = 6 元。
为了获得最大的满足感,森森决定用以下的方式坐地铁:在某一站上车(不妨设为地铁站 A),则对于所有车费相同的到达站,森森只会在计费距离最远的站或线路末端站点出站,然后用森森美图 App 在站点外拍一张认证照,再按同样的方式前往下一个站点。
坐着坐着,森森突然好奇起来:在给定出发站的情况下(在出发时森森也会拍一张照),他的整个旅程中能够留下哪些站点的认证照?
地铁是铁路运输的一种形式,指在地下运行为主的城市轨道交通系统。一般来说,地铁由若干个站点组成,并有多条不同的线路双向行驶,可类比公交车,当两条或更多条线路经过同一个站点时,可进行换乘,更换自己所乘坐的线路。举例来说,魔都 1 号线和 2 号线都经过人民广场站,则乘坐 1 号线到达人民广场时就可以换乘到 2 号线前往 2 号线的各个站点。换乘不需出站(也拍不到认证照),因此森森乘坐地铁时换乘不受限制。
输入格式:
输入第一行是三个正整数 N、M 和 K,表示魔都地铁有 N 个车站 (1 ≤ N ≤ 200),M 条线路 (1 ≤ M ≤ 1500),最短距离每超过 K 公里 (1 ≤ K ≤ 106),加 1 元车费。
接下来 M 行,每行由以下格式组成:
<站点1><空格><距离><空格><站点2><空格><距离><空格><站点3> ... <站点X-1><空格><距离><空格><站点X>
其中站点是一个 1 到 N 的编号;两个站点编号之间的距离指两个站在该线路上的距离。两站之间距离是一个不大于 106 的正整数。一条线路上的站点互不相同。
注意:两个站之间可能有多条直接连接的线路,且距离不一定相等。
再接下来有一个正整数 Q (1 ≤ Q ≤ 200),表示森森尝试从 Q 个站点出发。
最后有 Q 行,每行一个正整数 Xi**,表示森森尝试从编号为 **Xi 的站点出发。
输出格式:
对于森森每个尝试的站点,输出一行若干个整数,表示能够到达的站点编号。站点编号从小到大排序。
输入样例:
6 2 6
1 6 2 4 3 1 4
5 6 2 6 6
4
2
3
4
5
输出样例:
1 2 4 5 6
1 2 3 4 5 6
1 2 4 5 6
1 2 4 5 6
代码实现
#include <algorithm>
#include <cstdio>
#include <map>
#include <queue>
using namespace std;
const int maxn = 205;
const int INF = 0x3f3f3f3f;
int d[maxn][maxn];
int terminal[maxn], vis[maxn][maxn];
map<int, int> been[maxn];
int n, m, k;
int line[10000];int main() {
#ifdef LOCALfreopen("E:\\Cpp\\1.in", "r", stdin);
#endifscanf("%d%d%d", &n, &m, &k);for (int i = 1; i <= n; i++)for (int j = 1; j <= n; j++)d[i][j] = (i == j) ? 0 : INF;int u, v, len;int fare;char ch;while (m--) {int len = 0;while (scanf("%d", &u)) {line[len++] = u;ch = getchar();if (ch == '\n') {terminal[line[0]] = terminal[line[len - 1]] = 1;for (int i = 0; i != len - 1; i += 2) {u = line[i], v = line[i + 2];d[v][u] = d[u][v] = min(d[u][v], line[i + 1]);}break;}}}for (int k = 1; k <= n; k++) {for (int i = 1; i <= n; i++)for (int j = 1; j <= n; j++)d[i][j] = min(d[i][j], d[i][k] + d[k][j]);}for (int i = 1; i <= n; i++) {for (int j = 1; j <= n; j++) {if (i == j || d[i][j] == INF)continue;fare = 2 + d[i][j] / k;if (!been[i].count(fare) || been[i][fare] < d[i][j])been[i][fare] = d[i][j];}}int t, cur, first;queue<int> Q;scanf("%d", &t);while (t--) {first = 1;scanf("%d", &u);vis[u][u] = 1;Q.push(u);while (!Q.empty()) {cur = Q.front();Q.pop();for (int i = 1; i <= n; i++) {if (vis[u][i] || d[cur][i] == INF)continue;if (terminal[i]) {Q.push(i);vis[u][i] = 1;} else {fare = 2 + d[cur][i] / k;if (d[cur][i] == been[cur][fare]) {Q.push(i);vis[u][i] = 1;}}}}for (int i = 1; i <= n; i++) {if (vis[u][i]) {if (first) {printf("%d", i);first = 0;} elseprintf(" %d", i);}}printf("\n");}return 0;
}