제출 #1352200

#제출 시각아이디문제언어결과실행 시간메모리
1352200OwstinSirni (COCI17_sirni)C++20
140 / 140
1735 ms573840 KiB
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <utility>
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);
#define pb push_back
#define all(x) begin(x), end(x)
#define space " "
#define TEST_CASES int a; cin >> a; for (int i = 0; i < a; i++) {solve(); cout << endl;}
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef long double ld;
template<typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
mt19937 rng((uint32_t)chrono::steady_clock::now().time_since_epoch().count());

class DSU{
    vector<int> pred;
    vector<int> sizes;
public:
    DSU(int size) : pred(size), sizes(size, 1){
        for (int i = 0; i < size; i++){
            pred[i] = i;
        }
    }
    int size(int x){
        return (pred[x] == x ? sizes[x] : sizes[x] = sizes[find(x)]);
    }
    int find(int x){
        return (pred[x] == x ? x : pred[x] = find(pred[x]));
    }
    bool unite(int x, int y){
        int xp = find(x);
        int yp = find(y);
        if (xp == yp){
            return false;
        }
        if (sizes[xp] < sizes[yp]){
            swap(xp, yp);
        }
        sizes[xp] += sizes[yp];
        pred[yp] = xp;
        return true;
    }
};

void solve() {
    int n; cin >> n;
    vector<int> vec(n);
    for (int i = 0; i < n; i++) {
        cin >> vec[i];
    }
    sort(all(vec));
    vec.erase(unique(all(vec)), vec.end());
    n = vec.size();
    vector<int> close(vec.back() + 1, n);
    for (int i = 0; i < n; i++) {
        close[vec[i]] = i;
    }
    for (int i = vec.back() - 1; i >= 1; i--) {
        close[i] = min(close[i], close[i + 1]);
    }
    vector<vector<pair<int, int>>> edges(vec.back() + 1);
    for (int i = 0; i < n; i++) {
        for (int j = i + 1; j < n;) {
            edges[vec[j] % vec[i]].pb({i, j});
            int curr = vec[j] / vec[i];
            if ((curr + 1) * vec[i] <= vec.back()) {
                j = close[(curr + 1) * vec[i]];
            }
            else {
                break;
            }
        }
    }
    DSU dsu(n);
    ll res = 0;
    for (int i = 0; i <= vec.back(); i++) {
        for (auto x : edges[i]) {
            if (dsu.unite(x.first, x.second)) {
                res += i;
            }
        }
    }
    cout << res;
}

int main() {
    FAST_IO;
    //freopen("fencedin.in", "r", stdin);
    //freopen("fencedin.out", "w", stdout);
    //TEST_CASES;
    solve(); cout << endl;
    /*int a; cin >> a;
    for (int i = 1; i <= a; i++){
        cout << "Case #" << i << ": ";
        solve();
        cout << endl;
    }*/
}
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...