Submission #1339615

#TimeUsernameProblemLanguageResultExecution timeMemory
1339615Desh03Nile (IOI24_nile)C++20
0 / 100
33 ms6564 KiB
#include <bits/stdc++.h>

using namespace std;

struct DSU {
    vector<int> f, mn, ev, od;
    long long s;
    DSU(int n) {
        f.resize(n, -1);
        ev.resize(n, -1e9);
        od.resize(n, -1e9);
        mn.resize(n);
        iota(mn.begin(), mn.end(), 0);
        s = 0;
    }
    void set(int i, int x, bool c) {
        if (c) {
            od[i] = x;
        } else {
            ev[i] = x;
        }
    }
    int find(int u) {
        return f[u] < 0 ? u : f[u] = find(f[u]);
    }
    void unite(int u, int v) {
        u = find(u), v = find(v);
        if (u == v) return;
        if (f[u] > f[v]) swap(u, v);
        if (f[u] != -1 && (f[u] & 1)) {
            s -= ((mn[u] & 1) ? od[u] : ev[u]);
        }
        if (f[v] != -1 && (f[v] & 1)) {
            s -= ((mn[v] & 1) ? od[v] : ev[v]);
        }
        f[u] += f[v];
        f[v] = u;
        mn[u] = min(mn[u], mn[v]);
        od[u] = max(od[u], od[v]);
        ev[u] = max(ev[u], ev[v]);
        if (f[u] & 1) {
            s += ((mn[u] & 1) ? od[u] : ev[u]);
        }
    }
};

vector<long long> calculate_costs(vector<int> w, vector<int> a, vector<int> b, vector<int> e) {
    int n = w.size();
    vector<int> id(n);
    iota(id.begin(), id.end(), 0);
    sort(id.begin(), id.end(), [&](const int &i, const int &j) {
        return w[i] < w[j];
    });
    auto tw = w, ta = a, tb = b;
    for (int i = 0; i < n; i++) {
        w[i] = tw[id[i]];
        a[i] = ta[id[i]];
        b[i] = tb[id[i]];
    }
    vector<pair<int, int>> q;
    for (int i = 0; i + 1 < n; i++) {
        q.push_back({w[i + 1] - w[i], i});
    }
    sort(q.begin(), q.end());
    DSU ds(n);
    long long s = 0;
    for (int i = 0; i < n; i++) {
        s += a[i];
        ds.set(i, b[i] - a[i], (id[i] & 1));
    }
    int m = e.size();
    vector<pair<int, int>> ne(m);
    for (int i = 0; i < m; i++) {
        ne[i] = {e[i], i};
    }
    sort(ne.begin(), ne.end());
    vector<long long> ans(m);
    vector<bool> vis(n);
    for (int i = 0, j = 0; i < m; i++) {
        while (j + 1 < n && q[j].first <= ne[i].first) {
            ds.unite(q[j].second, q[j].second + 1);
            if (!vis[q[j].second]) {
                vis[q[j].second] = 1;
                s += b[q[j].second] - a[q[j].second];
            }
            if (!vis[q[j].second + 1]) {
                vis[q[j].second + 1] = 1;
                s += b[q[j].second + 1] - a[q[j].second + 1];
            }
            j++;
        }
        ans[ne[i].second] = s - ds.s;
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...