#include <bits/stdc++.h>
//#include "bits_stdc++.h"
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define endl '\n'
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define lb lower_bound
#define ub upper_bound
#define input(x) scanf("%d", &x);
#define input2(x, y) scanf("%lld%lld", &x, &y);
#define input3(x, y, z) scanf("%lld%lld%lld", &x, &y, &z);
#define input4(x, y, z, a) scanf("%lld%lld%lld%lld", &x, &y, &z, &a);
#define print(x, y) printf("%lld%c", x, y);
#define show(x) cerr << #x << " is " << x << endl;
#define show2(x,y) cerr << #x << " is " << x << " " << #y << " is " << y << endl;
#define show3(x,y,z) cerr << #x << " is " << x << " " << #y << " is " << y << " " << #z << " is " << z << endl;
#define all(x) x.begin(), x.end()
#define discretize(x) sort(x.begin(), x.end()); x.erase(unique(x.begin(), x.end()), x.end());
#define FOR(i, x, n) for (ll i =x; i<=n; ++i)
#define RFOR(i, x, n) for (ll i =x; i>=n; --i)
using namespace std;
mt19937_64 rnd(chrono::steady_clock::now().time_since_epoch().count());
//using namespace __gnu_pbds;
//#define ordered_set tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update>
//#define ordered_multiset tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update>
typedef int ll;
typedef long double ld;
typedef pair<ld, ll> pd;
typedef pair<string, ll> psl;
typedef pair<ll, ll> pi;
typedef pair<pi, ll> pii;
typedef pair<pi, pi> piii;
ll p;
vector<ll> dis;
bool compare(pii &a , pii &b)
{
return a.s < b.s;
}
ll par[100005], siz[100005];
vector<pi> bucket[10000005];
ll nxt[10000005];
inline ll find(ll x)
{
if (x==par[x]) return x;
return par[x] = find(par[x]);
}
inline void unite(ll x, ll y)
{
x = find(x), y = find(y);
if (x==y) return;
if (siz[x] < siz[y]) swap(x,y);
par[y] = x;
siz[x] += siz[y];
}
int main()
{
ll n;
long long ans = 0;
input(n);
dis.resize(n);
for (ll i=0; i<n; ++i) {input(dis[i]);}
discretize(dis);
for (ll i=0; i<= dis.back(); ++i)nxt[i] = -1;
for (ll i=0; i<dis.size(); ++i) nxt[dis[i]] = i;
for (ll i=dis.back(); i>=0; --i) if (nxt[i]==-1) nxt[i] = nxt[i+1];
for (ll i=0; i<dis.size(); ++i)
{
if (i + 1 < dis.size()) bucket[dis[i+1]%dis[i]].pb(mp(i, i+1));
for (ll j=dis[i]; j<=dis.back(); j+=dis[i])
{
bucket[dis[nxt[j]]%dis[i]].pb(mp(i, nxt[j]));
}
}
for (ll i=0; i<dis.size(); ++i) par[i] = i, siz[i] = 1;
ll cnt = 0;
for (ll i=0; i<dis.back()&& cnt < ll(dis.size())-1; ++i)
{
for (pi u: bucket[i])
{
if (find(u.f)==find(u.s)) continue;
ans += ll(i);
unite(u.f, u.s);
cnt++;
}
}
print(ans, '\n');
return 0;
}
Compilation message
sirni.cpp: In function 'int main()':
sirni.cpp:70:16: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
70 | for (ll i=0; i<dis.size(); ++i) nxt[dis[i]] = i;
| ~^~~~~~~~~~~
sirni.cpp:72:16: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
72 | for (ll i=0; i<dis.size(); ++i)
| ~^~~~~~~~~~~
sirni.cpp:74:19: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
74 | if (i + 1 < dis.size()) bucket[dis[i+1]%dis[i]].pb(mp(i, i+1));
| ~~~~~~^~~~~~~~~~~~
sirni.cpp:80:16: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
80 | for (ll i=0; i<dis.size(); ++i) par[i] = i, siz[i] = 1;
| ~^~~~~~~~~~~
sirni.cpp:12:23: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
12 | #define input(x) scanf("%d", &x);
| ~~~~~^~~~~~~~~~
sirni.cpp:65:2: note: in expansion of macro 'input'
65 | input(n);
| ^~~~~
sirni.cpp:12:23: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
12 | #define input(x) scanf("%d", &x);
| ~~~~~^~~~~~~~~~
sirni.cpp:67:26: note: in expansion of macro 'input'
67 | for (ll i=0; i<n; ++i) {input(dis[i]);}
| ^~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
163 ms |
274364 KB |
Output is correct |
2 |
Correct |
180 ms |
303696 KB |
Output is correct |
3 |
Correct |
142 ms |
274780 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
113 ms |
237204 KB |
Output is correct |
2 |
Correct |
777 ms |
669256 KB |
Output is correct |
3 |
Correct |
142 ms |
275028 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
137 ms |
274508 KB |
Output is correct |
2 |
Correct |
141 ms |
274428 KB |
Output is correct |
3 |
Correct |
138 ms |
274512 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
151 ms |
253036 KB |
Output is correct |
2 |
Correct |
196 ms |
281312 KB |
Output is correct |
3 |
Correct |
156 ms |
265908 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
111 ms |
243520 KB |
Output is correct |
2 |
Correct |
150 ms |
264764 KB |
Output is correct |
3 |
Correct |
142 ms |
250992 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
185 ms |
265188 KB |
Output is correct |
2 |
Correct |
207 ms |
298720 KB |
Output is correct |
3 |
Correct |
156 ms |
262316 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
124 ms |
242328 KB |
Output is correct |
2 |
Correct |
181 ms |
299684 KB |
Output is correct |
3 |
Correct |
137 ms |
263860 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
178 ms |
289932 KB |
Output is correct |
2 |
Correct |
677 ms |
634688 KB |
Output is correct |
3 |
Correct |
185 ms |
292624 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
178 ms |
294224 KB |
Output is correct |
2 |
Correct |
1060 ms |
747968 KB |
Output is correct |
3 |
Correct |
295 ms |
348696 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
145 ms |
277200 KB |
Output is correct |
2 |
Correct |
844 ms |
635564 KB |
Output is correct |
3 |
Correct |
159 ms |
266688 KB |
Output is correct |