题目
题意
这个题稍微有点绕(是我英语读题能力太差QAQ)。
- 最短路有多条的情况下,输出若干条最短路上与 0 相连接的最小序号的节点;
- 不能到达,输出
-1
; - 可以直接到达目标节点且该路径为最短路径,输出
0
。
题解
由于要输出最短路上与 0 相连接的最小序号的节点,直接来肯定比较麻烦,所以我们反向建图,这样最后的判断就会容易很多。
然后直接SPFA。
之后再判断一下是输出最小序号还是0
.
AC代码(注释更有料哦)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<string>
#include<queue>
#include<map>
#include<stack>
#include<list>
#include<set>
#include<deque>
#include<vector>
#include<ctime>
using namespace std;
//#pragma GCC optimize(2)
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define ull unsigned long long
#define ll long long
#define rep(i, x, y) for(int i=x;i<=y;i++)
#define mms(x, n) memset(x, n, sizeof(x))
#define mmc(A, tree) memcpy(A, tree, sizeof(tree))
#define eps (1e-8)
#define PI (acos(-1.0))
#define INF (0x3f3f3f3f)
#define mod (ull)(1e9+7)
typedef pair<int, int> P;
const int N = 1e5;
// 前式链向星存图
struct Node {
int to, w, ne;
} e[N];
int head[N], cnt = 1;
int dist[N];
bool vis[N];
int n, m;
queue<P> q; // 记录与0相连的节点
void init() {
cnt = 1;
mms(head, 0);
fill(dist, dist + N, INF);
mms(vis, false);
while (!q.empty()) q.pop();
}
void add(int u, int v, int w) {
e[cnt].w = w;
e[cnt].to = v;
e[cnt].ne = head[u];
head[u] = cnt++;
}
// 我只是一个普通的SPFA
void SPFA(int s) {
dist[s] = 0;
queue<int> q;
q.push(s);
vis[s] = true;
while (!q.empty()) {
int u = q.front();
q.pop();
vis[u] = false;
for (int i = head[u]; i; i = e[i].ne) {
int v = e[i].to, w = e[i].w;
if (dist[v] > dist[u] + w) {
dist[v] = dist[u] + w;
if (!vis[v]) {
q.push(v);
vis[v] = true;
}
}
}
}
}
int main() {
#ifndef ONLINE_JUDGE
freopen("input.txt", "r", stdin);
#endif
int T;
scanf("%d", &T);
while (T--) {
init();
scanf("%d%d", &n, &m);
rep(i, 1, m) {
int u, v, w;
scanf("%d%d%d", &u, &v, &w);
add(v, u, w);
if (u == 0) q.push(P(v, w));
}
SPFA(n + 1);
// 重点部分
if (dist[0] == INF) printf("-1\n"); // 无法到达的情况
else {
int ans = INF;
while (!q.empty()) {
P p = q.front();
q.pop();
int u = p.first, w = p.second;
if (dist[0] == (dist[u] + w)) { // 取出所有到0的距离与SPFA所得的最短距离相等的节点中序号最小的节点
ans = min(u, ans);
}
}
if (ans == n+1) printf("0\n"); // 如果该节点为n+1则说明直达且最短
else printf("%d\n", ans);
}
}
return 0;
}
后记
这个题真的有教育到我。
以前做题都是复制粘贴谷歌翻译一气呵成,真的打比赛的时候只能自己读题,很多地方都要读好几遍QAQ。
以后要自己读!