「AMPPZ2014」Petrol

Portal:
This is a bzoj permissions problem
Luogu team title link

Problem-solving ideas

First, for each point \ (X \) preconditioning the \ (nr [x] \) and \ (DIS [X] \) , respectively, from the \ (X \) nearest gas station, and the segment distance.
This process may use multi-source \ (\ text {Dijkstra} \ ) process.
Then, for each picture in the edge \ ((U, V, W) \) ( \ (NR [U] \ NE NR [V] \) )
to such an edge: \ ((NR [U], nr [v], dis [u
] + dis [v] + w) \) then just off with disjoint-set to maintain about connectivity.

Details Notes

• Do not write the shortest hang ah

Reference Code

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cctype>
#include <cmath>
#include <ctime>
#include <queue>
#define rg register
#define pii pair < int, int >
using namespace std;
template < typename T > inline void read(T& s) {
    s = 0; int f = 0; char c = getchar();
    while (!isdigit(c)) f |= (c == '-'), c = getchar();
    while (isdigit(c)) s = s * 10 + (c ^ 48), c = getchar();
    s = f ? -s : s;
}

const int _ = 200010;
const int __ = 400010;

int tot, head[_], nxt[__], ver[__], w[__];
inline void Add_edge(int u, int v, int d)
{ nxt[++tot] = head[u], head[u] = tot, ver[tot] = v, w[tot] = d; }

int q, n, m, k, c[_], vis[_], dis[_], nr[_];
struct node{ int u, v, d, id; }ask[_], g[__], e[__]; int ans[_];

inline bool cmp(const node& x, const node& y) { return x.d < y.d; }

inline void Dijkstra() {
    static priority_queue < pii > Q;
    memset(dis, 0x3f, sizeof dis);
    for (rg int i = 1; i <= k; ++i)
        Q.push(make_pair(dis[c[i]] = 0, c[i])), nr[c[i]] = c[i];
    while (!Q.empty()) {
        int u = Q.top().second; Q.pop();
        if (vis[u]) continue; vis[u] = 1;
        for (rg int i = head[u]; i; i = nxt[i]) {
            int v = ver[i];
            if (dis[v] > dis[u] + w[i])
                dis[v] = dis[u] + w[i], nr[v] = nr[u], Q.push(make_pair(-dis[v], v));
        }
    }
}

int fa[_];

inline int findd(int x) { return fa[x] == x ? x : fa[x] = findd(fa[x]); }

inline void merge(int x, int y) { fa[findd(x)] = findd(y); }

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.in", "r", stdin);
#endif
    read(n), read(k), read(m);
    for (rg int i = 1; i <= k; ++i) read(c[i]);
    for (rg int u, v, d, i = 1; i <= m; ++i)
        read(u), read(v), read(d), Add_edge(u, v, d), Add_edge(v, u, d), g[i] = (node) { u, v, d };
    Dijkstra();
    int cnt = 0;
    for (rg int u, v, i = 1; i <= m; ++i) {
        u = g[i].u, v = g[i].v;
        if (nr[u] != nr[v]) e[++cnt] = (node) { nr[u], nr[v], dis[u] + dis[v] + g[i].d };
    }
    read(q);
    for (rg int i = 1; i <= q; ++i)
        read(ask[i].u), read(ask[i].v), read(ask[i].d), ask[i].id = i;
    sort(ask + 1, ask + q + 1, cmp);
    sort(e + 1, e + cnt + 1, cmp);
    for (rg int i = 1; i <= n; ++i) fa[i] = i;
    for (rg int i = 1, j = 1; i <= q; ++i) {
        while (j <= cnt && e[j].d <= ask[i].d) merge(e[j].u, e[j].v), ++j;
        ans[ask[i].id] = findd(ask[i].u) == findd(ask[i].v);
    }
    for (rg int i = 1; i <= q; ++i) puts(ans[i] ? "TAK" : "NIE");
    return 0;
}

End Sahua \ (qwq \)