题目链接: [The Shortest Path in Nya Graph]
大致题意:
给定一幅有层次的线路图,第一行输入 N,M,C 表示 N 个几点和 M 条边,每两层之间通过的费用是 C。
意思就是,除了走给定的路之外,我们还可以选择穿透层次,花费 C 的费用走到上一层或者下一层的任意一个节点,而不去走题目给定的边。
然后第二行 N 个数就是表示第 iii 个节点的层号,然后 M 行是描述边的。
解题思路:
怎么样去维护层与层之间的连通,使他们相互通过花费为 C?在每层加两个虚拟的点,一个入点,一个出点,入点到每层的点权值为0,每层的点到出点权值为0,相邻层之间,出点与上一层或者下一侧的入点相连,权值为C,然后把m条额外的边加上,跑spfa
AC代码:
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
typedef long long ll;
using namespace std;
const int N = 3e5 + 5;
const int INF = 0x3f3f3f3f;
int vis[N];
vector<pair<int, int>>e[N];
int dis[N];
int a[N];
int n, m, c;
struct node {
int u, w;
node(int u = 0, int w = 0) :u(u), w(w) {}
bool operator <(const node& r)const {
return w > r.w;
}
};
void spfa() {
memset(vis, 0, sizeof vis);
memset(dis, INF, sizeof dis); dis[1] = 0;
priority_queue<node>q;
while (!q.empty())q.pop();
q.push(node(1, dis[1]));
while (!q.empty()) {
node tmp = q.top(); q.pop();
int u = tmp.u;
if (vis[u])continue;
vis[u] = 1;
for (int i = 0; i < e[u].size(); ++i) {
int v = e[u][i].first; int w = e[u][i].second;
if (dis[v] > dis[u] + w && !vis[v]) {
dis[v] = dis[u] + w;
q.push(node(v, dis[v]));
}
}
}
if (dis[n] == INF)printf("-1\n");
else printf("%d\n", dis[n]);
}
int main(void)
{
int k = 1;
int t; scanf("%d", &t);
while (t--) {
for (int i = 1; i <= 3 * n; ++i)e[i].clear();
scanf("%d%d%d", &n, &m, &c);
int x;
for (int i = 1; i <= n; ++i) {
scanf("%d", &x);
e[x * 2 + n - 1].push_back(make_pair(i, 0)); //入点
e[i].push_back(make_pair(x * 2 + n, 0)); //出点
}
for (int i = 1; i <= n; ++i) {
int u = 2 * i + n; //出点
if (i > 1) {
int v = u - 3; // 该层的出点,到上一层的入点
e[u].push_back(make_pair(v, c));
}
if (i < n) {
int v = u + 1; // 该层的出点,到下一层的入点
e[u].push_back(make_pair(v, c));
}
}
int u, v, w;
for (int i = 1; i <= m; ++i) {
scanf("%d%d%d", &u, &v, &w);
e[u].push_back(make_pair(v, w));
e[v].push_back(make_pair(u, w));
}
printf("Case #%d: ", k++);
spfa();
}
return 0;
}