题意
给定一个 n 个点 m 条边图,只能从点权高的点走到低的,且可以不计路程的瞬移至之前走过的某个点,求经过最多点的最短路径。
solution
求经过最多点显然直接bfs,建图的时候建高到低的单向边即可,(值得注意的是,若点权相同,则为相互可达的,需要建双向边)。然后根据bfs遍历可以走到的点,将走过的边加入边集,建一个新图出来。为了使路径最短,考虑最小生成树,但需要满足题目的条件,因此我们对新的图进行排序,以高度为第一关键字从大到小排,再以路径长度为第二关键字从小到大排,这样可以保证点最多的同时路径最短。
Code
#include <bits/stdc++.h>
using namespace std;
const int N = 2e6 + 5;
int n, m, tot, cnt, hi[N], h[N], pre[N], vis[N];
struct Node {
int from, to, v, next;
} g[N], E[N];
void add1(int x, int y, int z) {
g[tot].from = x, g[tot].to = y, g[tot].v = z, g[tot].next = h[x],
h[x] = tot++;
}
void add2(int x, int y, int z) {
cnt++, E[cnt].from = x, E[cnt].to = y, E[cnt].v = z;
}
bool cmp(Node x, Node y) {
if (hi[x.to] == hi[y.to]) return x.v < y.v;
return hi[x.to] > hi[y.to];
}
int xfind(int x) {
int i = x, j = x, temp;
while (i != pre[i]) i = pre[i];
while (i != j) {
temp = pre[j];
pre[j] = i;
j = temp;
}
return i;
}
void bfs() {
int res1 = 1;
queue<int> q;
vis[1] = true;
q.push(1);
while (q.size()) {
int u = q.front();
q.pop();
for (int i = h[u]; ~i; i = g[i].next) {
int v = g[i].to;
add2(u, v, g[i].v);
if (!vis[v]) {
vis[v] = true;
res1++;
q.push(v);
}
}
}
printf("%d ", res1);
}
void Kruscal() {
long long res2 = 0;
int num = 0;
for (int i = 1; i <= cnt; i++) {
int x = E[i].from, y = E[i].to;
int fx = xfind(x), fy = xfind(y);
if (fx != fy) {
pre[fx] = fy;
res2 += E[i].v;
num++;
}
if (num == n - 1) break;
}
printf("%lld\n", res2);
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &hi[i]);
pre[i] = i;
}
memset(h, -1, sizeof(h));
for (int i = 1; i <= m; i++) {
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
if (hi[x] >= hi[y]) add1(x, y, z);
if (hi[x] <= hi[y]) add1(y, x, z);
}
bfs();
sort(E + 1, E + cnt + 1, cmp);
Kruscal();
return 0;
}
京公网安备 11010502036488号