Submission #307550

#	Time	Username	Problem	Language	Result	Execution time	Memory
307550		luciocf	Sirni (COCI17_sirni)	C++14	84 / 140	5125 ms	527632 KiB

sirni

#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;
 
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 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 m = 0;
	dsu.init(n);

	vector<pii> V;
 
	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 = a[i]; v <= a[n]; v += a[i])
			V.push_back({v, i});
	}
 
	sort(V.begin(), V.end());
 
	set<pair<int, pii>> st;
 
	for (int i = 2; i <= n; i++)
		st.insert({a[i]-a[i-1], {i-1, i}});
 
	int ptr = (int)V.size()-1;
 
	for (int i = n; i >= 1; i--)
	{
		while (ptr >= 0 && V[ptr].ff > a[i])
			ptr--;
 
		while (ptr >= 0 && V[ptr].ff > a[i-1])
		{
			st.insert({a[i]-V[ptr].ff, {i, V[ptr].ss}});
			ptr--;
		}
	}
 
 
	ll ans = 0;
 
	for (auto pp: st)
	{
		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 (stderr)

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

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