제출 #86486

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
86486		dimash241	Sirni (COCI17_sirni)	C++14	112 / 140	5139 ms	616860 KiB

sirni

#include <stdio.h>
# include <bits/stdc++.h> 
 
#define _USE_MATH_DEFINES_
#define ll long long
#define ld long double
#define Accepted 0
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x.size())
#define every(x) x.begin(),x.end()
#define F first
#define S second
#define For(i,x,y)  for (ll i = x; i <= y; i ++) 
#define FOr(i,x,y)  for (ll i = x; i >= y; i --)
#define SpeedForce ios_base::sync_with_stdio(0), cin.tie(0)
// ROAD to...                                                                                                                                                                                                                Red
 
using namespace std;     
const int maxn = 1e7 + 3;
const int N = 1e5 + 1;
 
int n;
int sz[N];
int p[N];
int a[N];
int R[maxn + 5];
int id[maxn + 5];
int ans = 0;                                  
int u[maxn + 5];
vector < pair < int, int > > g[maxn + 5];
 
int get (int v) {
	if (p[v] == v) {
		return v;
	}
 
	return p[v] = get(p[v]);
}
 
bool uni (int u, int v) {
	u = get(u), v = get(v);
	if (u == v) return 0;
 
	if (sz[u] > sz[v]) swap(u, v);
	p[u] = v;
	if (sz[v] == sz[u]) sz[v] ++;
 
	return 1;
}
 
int main () {
	SpeedForce;               
    cin >> n;
    For (i, 1, n) {
    	cin >> a[i];
    }

    For (i, 1, n) {
    	if (id[a[i]]) continue;
    	p[i] = i;
    	sz[i] = 1;
    	R[a[i]] = a[i];
    	id[a[i]] = i;
    }

    for (int j = maxn - 2; j >= 0; j --) {
    	if (!R[j])
    		R[j] = R[j + 1];
    }

    For (i, 1, n) {
    	if (u[a[i]] ++ > 0) continue;
    	if (a[i] * 2 >= maxn - 1 || R[a[i] + 1] != R[a[i] * 2]) {
    		g[R[a[i] + 1] % a[i]].pb(mp(i, id[R[a[i] + 1]]));
    	}
    	for (int j = a[i]; j < maxn && R[j]; j += a[i]) {
    		if (j + a[i] > maxn || R[j] != R[j + a[i]]) {
    			g[R[j] % a[i]].pb(mp(i, id[R[j]]));
    		}
    	}
    	  
    }
	for (int i = 0; i < maxn; i ++) {
   		if (g[i].empty()) continue;
   		for (auto it : g[i]) {
   			if (uni (it.F, it.S)) {
   				ans += i;
   			}
   		}
   	}
 
    cout << ans;
    
    return Accepted;
}
 
// Coded By OB

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 14 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 14 / 14
• Subtask 9: 0 / 14
• Subtask 10: 0 / 14