Submission #1118339

#	Time	Username	Problem	Language	Result	Execution time	Memory
1118339		gdragon	Evacuation plan (IZhO18_plan)	C++17	35 / 100	431 ms	27960 KiB

plan

#include <bits/stdc++.h>
using namespace std;
#define TASK "long"
#define fi first
#define se second
#define ll long long
#define pb push_back
#define ALL(x) (x).begin(), (x).end()
#define GETBIT(mask, i) ((mask) >> (i) & 1)
#define MASK(i) ((1LL) << (i))
#define SZ(x) ((int)(x).size())
#define mp make_pair
#define CNTBIT(mask) __builtin_popcount(mask)
template<class X, class Y> bool maximize(X &x, Y y){ if (x < y) {x = y; return true;} return false;};
template<class X, class Y> bool minimize(X &x, Y y){ if (x > y) {x = y; return true;} return false;};
typedef pair<int, int> ii;
const int N = 1e5 + 5;
const int M = 5e5 + 5;
const int inf = 1e9 + 7;
const long long INF = (long long)1e18 + 7;
const int mod = 1e9 + 7;
vector<ii> adj[N];
int n, m, q, numNpp;
vector<int> Npps, val;
int d[N];
void read() {
    cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int u, v, c; cin >> u >> v >> c;
        adj[u].push_back({v, c});
        adj[v].push_back({u, c});
    }
    cin >> numNpp;
    Npps.assign(numNpp, 0);
    for(int &i: Npps) cin >> i;
}
void calDistToNpp() {
    priority_queue<ii, vector<ii>, greater<ii>> heap;
    memset(d, 0x3f, sizeof(d));
    for(int i: Npps) {
        heap.push({0, i});
        d[i] = 0;
    }
    while(!heap.empty()) {
        ii tmp = heap.top(); heap.pop();
        int u = tmp.se;
        int du = tmp.fi;
        if (du != d[u]) continue;
        for(auto [v, c]: adj[u]) {
            if (minimize(d[v], d[u] + c)) {
                heap.push(mp(d[v], v));
            }
        }
    }
}
bool vis[N];
bool dfs(int u, int t, int lim) {
    if (vis[u]) return false;
    vis[u] = 1;
    if (d[u] < lim) return false;
    if (u == t) return true;
    for(auto [v, c]: adj[u]) if (d[v] >= lim) {
        if (dfs(v, t, lim)) return true;
    } 
    return false;
}
bool check(int s, int t, int lim) {
    fill(vis + 1, vis + n + 1, 0);
    return dfs(s, t, lim);
}
void solve() {
    calDistToNpp();
    for(int i = 1; i <= n; i++) val.push_back(d[i]);
    sort(ALL(val));
    val.erase(unique(ALL(val)), val.end());
    cin >> q;
    while(q--) {
        int s, t; cin >> s >> t;
        if (max(n, q) > 1e3) {
            cout << min(d[s], d[t]) << endl;
            continue;
        }
        int l = 0, r = SZ(val) - 1, ans = 0;
        while(l <= r) {
            int mid = (l + r) >> 1;
            if (check(s, t, val[mid])) {
                ans = val[mid];
                l = mid + 1;
            }
            else r = mid - 1;
        } 
        cout << ans << endl;
    }
}
signed main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    if (fopen(TASK".inp", "r")) {
        freopen(TASK".inp", "r", stdin);
        freopen(TASK".out", "w", stdout);
    }
    int test = 1;
    // cin >> test;
    while(test--) {
        read();
        solve();
    }
    return 0;
}

// rmq - rmq2d
// hash
// fw - fw2d
// segtree

Compilation message (stderr)

plan.cpp: In function 'int main()':
plan.cpp:98:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   98 |         freopen(TASK".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
plan.cpp:99:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   99 |         freopen(TASK".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 13 / 13
• Subtask 3: 12 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 46