oj.uz

답안 #307573

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
307573 2020-09-28T16:08:06 Z luciocf Sirni (COCI17_sirni) C++17
112 / 140
 5000 ms 408952 KB 
#include <bits/stdc++.h>
 
#define ff first
#define ss second
 
using namespace std;
 
typedef long long ll;
typedef pair<int, int> pii;
 
const int maxn = 1e5+10;
const int maxv = 1e7+1;
 
struct DSU
{
	int pai[maxn], peso[maxn];
 
	void init(int n)
	{
		for (int i = 1; i <= n; i++)
			pai[i] = i, peso[i] = 1;
	}
 
	int Find(int x)
	{
		if (pai[x] == x) return x;
		return pai[x] = Find(pai[x]);
	}
 
	void Join(int x, int y)
	{
		x = Find(x), y = Find(y);
		if (x == y) return;
 
		if (peso[x] < peso[y]) swap(x, y);
 
		pai[y] = x, peso[x] += peso[y];
	}
} dsu;
 
int a[maxn];
int prim[maxv];
 
int main(void)
{
	int n;
	scanf("%d", &n);
 
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
 
	sort(a+1, a+n+1);
 		
	int tot = 0;
	for (int i = 1; i <= n; i++)
	{
		if (a[i] == a[i+1]) continue;
		tot += (a[n]/a[i]);
	}

 	int ptr = 2;
	for (int i = a[1]; i <= a[n]; i++)
	{
		while (ptr <= n && a[ptr] < i)
			ptr++;
 
		prim[i] = ptr;
	}
 
 	int m = 0;
	dsu.init(n);
 
	vector<pair<int, pii>> V;
	V.reserve(tot+1);
 
	for (int i = 2; i <= n; i++)
		V.push_back({a[i]-a[i-1], {i-1, i}});
 
	for (int i = 1; i <= n; i++)
	{
		int j;
		for (j = i; j <= n && a[i] == a[j]; j++)
			if (j != i)
				dsu.Join(i, j), m++;
 
		i = j-1;
			
		for (int v = 2*a[i]; v <= a[n]; v += a[i])
		{
			j = v;
			for (; j <= a[n] && prim[j] == prim[v]; j += a[i])
				;
			j -= a[i];

			v = j;

			if (a[prim[v]] == v)
			{
				if (dsu.Find(prim[v]) != dsu.Find(i))
				{
					++m;
					dsu.Join(prim[v], i);
				}
			}
			else V.push_back({a[prim[v]]-v, {i, prim[v]}});
		}
	}
 
 	sort(V.begin(), V.end());
 
	ll ans = 0;
 
	for (auto pp: V)
	{
		if (m == n-1) break;
 
		int u = pp.ss.ff, v = pp.ss.ss, w = pp.ff;
 
		if (dsu.Find(u) != dsu.Find(v))
		{
			ans += 1ll*w;
			dsu.Join(u, v), ++m;
		}
	}
 
	printf("%lld\n", ans);
}

Compilation message

sirni.cpp: In function 'int main()':
sirni.cpp:47:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   47 |  scanf("%d", &n);
      |  ~~~~~^~~~~~~~~~
sirni.cpp:50:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   50 |   scanf("%d", &a[i]);
      |   ~~~~~^~~~~~~~~~~~~

Subtask #1
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 35 ms 39416 KB Output is correct
2 Correct 99 ms 41692 KB Output is correct
3 Correct 35 ms 39416 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 512 KB Output is correct
2 Correct 287 ms 40184 KB Output is correct
3 Correct 36 ms 39552 KB Output is correct

Subtask #3
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 34 ms 39416 KB Output is correct
2 Correct 33 ms 39356 KB Output is correct
3 Correct 34 ms 39592 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 199 ms 16376 KB Output is correct
2 Correct 653 ms 46712 KB Output is correct
3 Correct 257 ms 22008 KB Output is correct

Subtask #5
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 29 ms 6016 KB Output is correct
2 Correct 381 ms 31224 KB Output is correct
3 Correct 184 ms 14456 KB Output is correct

Subtask #6
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 418 ms 30072 KB Output is correct
2 Correct 848 ms 57592 KB Output is correct
3 Correct 239 ms 19704 KB Output is correct

Subtask #7
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 56 ms 5120 KB Output is correct
2 Correct 798 ms 56440 KB Output is correct
3 Correct 231 ms 19772 KB Output is correct

Subtask #8
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 309 ms 54648 KB Output is correct
2 Correct 4814 ms 303864 KB Output is correct
3 Correct 355 ms 57912 KB Output is correct

Subtask #9
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 292 ms 54628 KB Output is correct
2 Execution timed out 5056 ms 408568 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 72 ms 41848 KB Output is correct
2 Execution timed out 5119 ms 408952 KB Time limit exceeded
3 Halted 0 ms 0 KB -