Submission #1006301

#TimeUsernameProblemLanguageResultExecution timeMemory
1006301orcslopSirni (COCI17_sirni)C++17
140 / 140
989 ms766628 KiB
#include <bits/stdc++.h> using namespace std; using ll = long long; // #define int long long #define pb push_back #define all(x) begin(x), end(x) #define sz(x) (int) (x).size() #define f first #define s second #define mkp make_pair #define pii pair<int, int> struct DSU { vector<int> e; DSU(int N) { e = vector<int>(N, -1); } int get(int x) { return e[x] < 0 ? x : e[x] = get(e[x]); } bool same_set(int a, int b) { return get(a) == get(b); } int size(int x) { return -e[get(x)]; } bool unite(int x, int y) { x = get(x), y = get(y); if (x == y) return false; if (e[x] > e[y]) swap(x, y); e[x] += e[y]; e[y] = x; return true; } }; int32_t main() { cin.tie(0)->sync_with_stdio(0); int n; cin >> n; vector<int> v(n); for(int i = 0; i < n; i++){ cin >> v[i]; } sort(all(v)); v.erase(unique(all(v)), v.end()); n = sz(v); vector<int> nxt(v.back() + 1, -1); for(int i = 0; i < n; i++){ nxt[v[i]] = i; } for(int i = v.back() - 1; i >= 0; i--){ if(nxt[i] == -1){ nxt[i] = nxt[i + 1]; } } vector<vector<pii>> links(v.back() + 1); for(int i = 0; i < n - 1; i++){ links[v[i + 1] % v[i]].pb(mkp(i, i + 1)); for(int j = 2 * v[i]; j <= v.back(); j += v[i]){ links[v[nxt[j]] % v[i]].pb(mkp(nxt[j], i)); } } DSU dsu(n); ll ans = 0; for(int i = 0; i <= v.back(); i++){ for(auto x : links[i]){ if(dsu.unite(x.f, x.s)){ ans += 1LL * i; } } } cout << ans << '\n'; return 0; }
#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...