oj.uz

답안 #1020071

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1020071 2024-07-11T13:30:28 Z adrielcp Sirni (COCI17_sirni) C++17
140 / 140
 1033 ms 747280 KB 
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define lld double
// #define int ll
#define usaco(fname) freopen(#fname ".in","r",stdin);freopen(#fname ".out","w",stdout);
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }
// const ll INF = 1e18;
const int INF = 1e9;
const int mod = 1e9+7;
const lld PI = acos(-1.0);
int di[] = {1, -1, 0, 0, 1, 1, -1, -1};
int dj[] = {0, 0, 1, -1, 1, -1, 1, -1};
#define debug(x) cout << #x << ": " << x << endl;
#define add(a, b) a += b, a %= mod
#define mul(a, b) ((a % mod) * (b % mod)) % mod
#define all(x) x.begin(),x.end()

struct DSU {
  vector<int> p, sz, mn, mx; 
  DSU(int n) {
    p = vector<int>(n);
    // sz = vector<int>(n, 1);
    // mn = vector<int>(n);
    for(int i = 0; i < n; i++) p[i] = i;
  }

  int find(int a) {
    if (a == p[a]) return p[a];
    return p[a] = find(p[a]);
  }

  int join(int a, int b) {
    int r1 = find(a);
    int r2 = find(b);
    if (r1 == r2) return 0;
    p[r1] = r2;
    // sz[r2] += sz[r1];
    return 1;
  }
};

const int N = 1e7+69;
vector<pair<int, int>> bruh[N];
int nxt[N];

const int D = 1e5+5;
int a[D];

void solve() {
  int _n;cin>>_n;
  // vector<int> a(_n);
  // DSU dsu(n);
  for (int i = 0; i < _n; i++) cin >> a[i];
  sort(a, a+_n);
  fill(nxt, nxt+N, -1);

  // vector<vector<pair<int,int>>> bruh(N,vector<pair<int,int>>());

  // vector<int> nxt(N, -1);

  int n = unique(a, a+_n) - a;

  int mx = 0;
  for (int i = 0; i < n; i++) {
    nxt[a[i]] = i;
    mx = max(mx, a[i]);
  }

  // int mx = *max_element(all(a));
  for (int i = mx-1; i >= 0; i--) {
    if (nxt[i] == -1) nxt[i] = nxt[i+1];
  }
  
  for (int i = 0; i+1 < n; i++) {
    bruh[a[i+1]%a[i]].push_back({i+1, i});
    for (int j = a[i]; j <= mx; j+=a[i]) {
      bruh[a[nxt[j]]%a[i]].push_back({nxt[j], i});
    }
  }

  DSU dsu(n);

  long long ans = 0;
  for (int i = 0; i < mx; i++) {
    for (auto &[u, v] : bruh[i]) {
      if (dsu.join(u, v)) ans += i;
    }
  }

  // // for (auto [x, y, z] : bruh) cout << x << " " << y << " " << z << endl;
  // int ans = 0;
  cout << ans << endl;
}

int32_t main() {
  ios_base::sync_with_stdio(0);cin.tie(0);
  int t=1;
  while(t--) solve();

  return 0;
}

Subtask #1
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 121 ms 274512 KB Output is correct
2 Correct 176 ms 303696 KB Output is correct
3 Correct 125 ms 274768 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 102 ms 274444 KB Output is correct
2 Correct 848 ms 668952 KB Output is correct
3 Correct 123 ms 275240 KB Output is correct

Subtask #3
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 135 ms 274512 KB Output is correct
2 Correct 123 ms 274256 KB Output is correct
3 Correct 122 ms 274584 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 138 ms 285828 KB Output is correct
2 Correct 183 ms 313852 KB Output is correct
3 Correct 148 ms 298560 KB Output is correct

Subtask #5
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 108 ms 276764 KB Output is correct
2 Correct 154 ms 298188 KB Output is correct
3 Correct 165 ms 286052 KB Output is correct

Subtask #6
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 158 ms 298456 KB Output is correct
2 Correct 207 ms 331488 KB Output is correct
3 Correct 178 ms 294832 KB Output is correct

Subtask #7
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 116 ms 278464 KB Output is correct
2 Correct 209 ms 331960 KB Output is correct
3 Correct 151 ms 296376 KB Output is correct

Subtask #8
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 170 ms 289480 KB Output is correct
2 Correct 813 ms 633956 KB Output is correct
3 Correct 174 ms 292116 KB Output is correct

Subtask #9
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 171 ms 293664 KB Output is correct
2 Correct 1033 ms 747280 KB Output is correct
3 Correct 272 ms 348564 KB Output is correct

Subtask #10
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 132 ms 277068 KB Output is correct
2 Correct 868 ms 634980 KB Output is correct
3 Correct 150 ms 299804 KB Output is correct