oj.uz

Submission #1085122

# Submission time Handle Problem Language Result Execution time Memory
1085122 2024-09-07T14:54:23 Z fryingduc Sirni (COCI17_sirni) C++17
84 / 140
 1010 ms 786432 KB 
#include "bits/stdc++.h"
using namespace std;

#ifdef duc_debug
#include "bits/debug.h"
#else 
#define debug(...)     
#endif

const int maxn = 1e5 + 5;
const int N = 1e7 + 7;
int n;
vector<int> value[N];

int lab[maxn];
int nxt[N];

int find(int u) {
  return lab[u] < 0 ? u : lab[u] = find(lab[u]);
}
bool is_joined(int u, int v) {
  u = find(u), v = find(v);
  if(u == v) return 1;
  if(lab[u] > lab[v]) swap(u, v);
  lab[u] += lab[v];
  lab[v] = u;
  return 0;
}
struct edge {
  int u, v, w;
  bool operator<(const edge &o) {
    return w < o.w;
  }
};
vector<edge> e;
void solve() {
  cin >> n;
  vector<int> a;
  for(int i = 1; i <= n; ++i) {
    int x; cin >> x;
    a.push_back(x);
    lab[i] = -1;
    value[x].push_back(i);
  }
  for(int i = N - 2; i; --i) {
    nxt[i] = (value[i].empty() ? nxt[i + 1] : i);
  }
  sort(a.begin(), a.end());
  a.erase(unique(a.begin(), a.end()), a.end());
  long long ans = 0;
  for(int i = 1; i < (int)a.size(); ++i) {
    e.push_back({value[a[i]][0], value[a[i - 1]][0], a[i] % a[i - 1]});
  }
  for(int i = 0; i < (int)a.size(); ++i) {
    for(int j = a[i]; j <= a.back(); j += a[i]) {
      int val = nxt[j];
      for(auto pos:value[val]) {
        e.push_back({value[a[i]][0], pos, val % a[i]});
      }
    }
  }
  sort(e.begin(), e.end());
  for(auto i:e) {
    if(!is_joined(i.u, i.v)) {
      ans += i.w;
    }
  }
  cout << ans;
}
signed main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  solve();

  return 0;
}

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 121 ms 274516 KB Output is correct
2 Correct 281 ms 323744 KB Output is correct
3 Correct 124 ms 274876 KB Output is correct

Subtask #2
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 123 ms 274512 KB Output is correct
2 Runtime error 720 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 124 ms 274512 KB Output is correct
2 Correct 121 ms 274520 KB Output is correct
3 Correct 122 ms 274572 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 252 ms 303108 KB Output is correct
2 Correct 558 ms 376548 KB Output is correct
3 Correct 333 ms 327200 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 137 ms 277960 KB Output is correct
2 Correct 309 ms 325556 KB Output is correct
3 Correct 234 ms 302552 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 322 ms 327572 KB Output is correct
2 Correct 565 ms 376800 KB Output is correct
3 Correct 305 ms 327724 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 157 ms 282016 KB Output is correct
2 Correct 540 ms 376216 KB Output is correct
3 Correct 317 ms 327392 KB Output is correct

Subtask #8
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 278 ms 303196 KB Output is correct
2 Runtime error 1010 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #9
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 299 ms 303260 KB Output is correct
2 Runtime error 940 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 146 ms 278424 KB Output is correct
2 Runtime error 1010 ms 786432 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -