oj.uz

Submission #1021302

# Submission time Handle Problem Language Result Execution time Memory
1021302 2024-07-12T16:45:58 Z pan Sirni (COCI17_sirni) C++17
140 / 140
 1060 ms 747968 KB 
#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]);}
      |                          ^~~~~

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory 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

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory 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