Submission #86534

#	Time	Username	Problem	Language	Result	Execution time	Memory
86534		dimash241	Sirni (COCI17_sirni)	C++14	112 / 140	5137 ms	608760 KiB

sirni

#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,sse3,sse4,popcnt,abm,mmx")
 
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <string>
#include <bitset>
#include <cstdio>
#include <limits>
#include <vector>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <fstream>
#include <numeric>
#include <sstream>
#include <cassert>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <unordered_set>
#include <unordered_map>
 
#define pb push_back
#define F first
#define S second
#define For(i,x,y)  for (int i = x; i <= y; ++i) 
#define FOr(i,x,y)  for (int i = x; i >= y; --i)
// 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 + 1];
int id[maxn + 1];
int ans = 0;                                  
bool u[maxn + 1];
vector < pair < int, int > > g[maxn + 1];
 
inline int get (int v) {
	if (p[v] == v) {
		return v;
	}
 
	return p[v] = get(p[v]);
}
 
int main () {               
    scanf("%d", &n);
    For (i, 1, n) {
    	scanf("%d", a + i);
    }
    sort(a + 1, a + n + 1);
    if (a[1] == 1) {
    	puts("0");
    	exit(0);
    }
 
    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;
    	++ u[a[i]];
    	if (a[i] + a[i] >= maxn - 1 || R[a[i] + 1] != R[a[i] + a[i]]) {
    		g[R[a[i] + 1] % a[i]].pb({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] - j].pb({i, id[R[j]]});
    		}
    	}
    	  
    }
 
	for (int i = 0; i < maxn; ++i) {
   		for (auto it : g[i]) {
   			it.F = get(it.F);
   			it.S = get(it.S);
   			if (it.F == it.S) continue;
   			ans += i;
   			if (sz[it.F] > sz[it.S]) {
   				p[it.S] = it.F;
   			} else if (sz[it.F] < sz[it.S]) {
   				p[it.F] = it.S;
   			} else {
   				p[it.F] = it.S;
   				sz[it.F] ++;
   			}
   		}
   	}
 
    printf("%d", ans);   
}

Compilation message (stderr)

sirni.cpp: In function 'int main()':
sirni.cpp:82:15: warning: use of an operand of type 'bool' in 'operator++' is deprecated [-Wdeprecated]
      ++ u[a[i]];
               ^
sirni.cpp:57:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
sirni.cpp:59:11: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
      scanf("%d", a + i);
      ~~~~~^~~~~~~~~~~~~

• 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