Submission #734513

#	Time	Username	Problem	Language	Result	Execution time	Memory
734513		TahirAliyev	Sirni (COCI17_sirni)	C++17	112 / 140	2850 ms	786432 KiB

sirni

#include <bits/stdc++.h>
 
#pragma GCC optimize("O3")
 
using namespace std;
 
#define ll long long int
#define oo 1e9
#define pii pair<ll, int>



const int MAX = 1e5 + 5;
int n;

int par[MAX];

int findSet(int u){
    if(par[u] < 0) return u;
    return par[u] = findSet(par[u]);
}

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

bool check(int u, int v){
    u = findSet(u);
    v = findSet(v);
    if(u == v) return true;
    return false;
}

vector<pii> edges[int(1e7 + 7)];

void solve(){
    memset(par, -1, sizeof(par));
    cin >> n;
    vector<int> v;
    for (int i = 0; i < n; i++)
    {
        int a; cin >> a;
        v.push_back(a);
    }
    sort(v.begin(), v.end());
    int m = v[v.size() - 1];
    for (int i = 0; i < n; i++)
    {
        if(i != 0 && v[i] == v[i - 1]) continue;
        for (int r = v[i]; r <= m; r += v[i])
        {
            auto itr = lower_bound(v.begin(), v.end(), r);
            if(r == v[i]){
                itr = upper_bound(v.begin(), v.end(), r);
            }
            if(itr == v.end()){
                break;
            }
            if((*itr) >= (r + v[i])){
                r = (*itr) / v[i] * v[i] - v[i];
                continue;
            }
            edges[(*itr) % v[i]].push_back({i, int(itr - v.begin())});
        }
    }
    ll ans = 0;
    for(int i = 0; i <= 1e7; ++i){
        for(auto& p:edges[i]){
            if(unite(p.first, p.second)){
                ans += i;
            }
        }
    }
    cout << ans;
}

int main(){
    solve();
}

• 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: 0 / 14
• Subtask 10: 0 / 14