Submission #1109879

#	Time	Username	Problem	Language	Result	Execution time	Memory
1109879		juliany2	Sirni (COCI17_sirni)	C++17	140 / 140	1099 ms	581460 KiB

sirni

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define all(x) (x).begin(), (x).end()

struct DSU {
    vector<int> e;

    DSU(int sz) { e = vector<int> (sz + 1, -1); }

    int get(int x) { return e[x] < 0 ? x : e[x] = get(e[x]); }

    bool unite(int u, int v) {
        u = get(u), v = get(v);
        if (u == v) return false;
        if (e[v] < e[u]) swap(u, v);
        e[u] += e[v], e[v] = u;
        return true;
    }
};

const int M = 1e7 + 7;
int n;
int nxt[M];
vector<array<int, 2>> e[M];

int main() {
    cin.tie(0)->sync_with_stdio(false);

    memset(nxt, -1, sizeof(nxt));

    cin >> n;

    vector<int> a(n);
    for (int &i : a)
        cin >> i;

    sort(all(a));
    a.erase(unique(all(a)), a.end());

    n = a.size();
    for (int i = 0; i < n; i++) {
        nxt[a[i]] = i;
    }

    for (int i = M - 2; i >= 0; i--) {
        if (nxt[i] == -1)
            nxt[i] = nxt[i + 1];
    }

    for (int i = 0; i < n; i++) {
        vector<int> l;

        if (~nxt[a[i] + 1])
            l.push_back(nxt[a[i] + 1]);

        for (int j = a[i] * 2; j < M; j += a[i]) {
            if (nxt[j] == -1)
                break;
            l.push_back(nxt[j]);
        }

        l.erase(unique(all(l)), l.end());

        for (int j : l) {
            e[a[j] % a[i]].push_back({i, j});
        }
    }

    DSU dsu(n);

    ll ans = 0;
    for (int i = 0; i < M; i++) {
        for (auto &[u, v] : e[i]) {
            if (dsu.unite(u, v)) {
                ans += i;
            }
        }

        if (-dsu.e[dsu.get(1)] == n)
            break;
    }

    cout << ans << '\n';

    return 0;
}

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 14 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 14 / 14
• Subtask 9: 14 / 14
• Subtask 10: 14 / 14