oj.uz

Submission #366067

# Submission time Handle Problem Language Result Execution time Memory
366067 2021-02-13T01:03:55 Z mahmoudbadawy Sjekira (COCI20_sjekira) C++17
110 / 110
 275 ms 10348 KB 
#include <bits/stdc++.h>
#define F first
#define S second

using namespace std;

const int N=1e5+5;

set< pair<int, pair<int,int> > > ss;
int uni[N],mx[N];
int n;

int uni_find(int x)
{
	return uni[x]=(uni[x]==x?x:uni_find(uni[x]));
}

int unio(int x,int y)
{
	x=uni_find(x); y=uni_find(y);
	if(x==y) return 0;
	uni[x]=y;
	mx[y]=max(mx[x],mx[y]);
	return 1;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		uni[i]=i;
		scanf("%d",&mx[i]);
	}
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d %d",&x,&y);
		ss.insert({mx[x]+mx[y],{x,y}});
	}
	long long ans=0;
	while(ss.size())
	{
		auto cur=*ss.begin(); ss.erase(ss.begin());
		int a=uni_find(cur.S.F),b=uni_find(cur.S.S);
		if(a==b) continue;
		if(mx[a]+mx[b]!=cur.F)
			ss.insert({mx[a]+mx[b],{cur.S.F,cur.S.S}});
		else
		{
			ans+=mx[a]+mx[b];
			unio(a,b);
		}
	}
	printf("%lld\n",ans);
}

Compilation message

sjekira.cpp: In function 'int main()':
sjekira.cpp:29:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   29 |  scanf("%d",&n);
      |  ~~~~~^~~~~~~~~
sjekira.cpp:33:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   33 |   scanf("%d",&mx[i]);
      |   ~~~~~^~~~~~~~~~~~~
sjekira.cpp:38:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   38 |   scanf("%d %d",&x,&y);
      |   ~~~~~^~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct
4 Correct 1 ms 364 KB Output is correct
5 Correct 1 ms 364 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 116 ms 7788 KB Output is correct
2 Correct 94 ms 5996 KB Output is correct
3 Correct 75 ms 5740 KB Output is correct
4 Correct 90 ms 6508 KB Output is correct
5 Correct 208 ms 9708 KB Output is correct
6 Correct 74 ms 10220 KB Output is correct
7 Correct 62 ms 8684 KB Output is correct
8 Correct 55 ms 8044 KB Output is correct
9 Correct 38 ms 5484 KB Output is correct
10 Correct 70 ms 10348 KB Output is correct

Subtask #3
30.0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct
4 Correct 1 ms 364 KB Output is correct
5 Correct 1 ms 364 KB Output is correct
6 Correct 2 ms 364 KB Output is correct
7 Correct 2 ms 364 KB Output is correct
8 Correct 1 ms 364 KB Output is correct
9 Correct 2 ms 364 KB Output is correct
10 Correct 2 ms 364 KB Output is correct

Subtask #4
60.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct
4 Correct 1 ms 364 KB Output is correct
5 Correct 1 ms 364 KB Output is correct
6 Correct 116 ms 7788 KB Output is correct
7 Correct 94 ms 5996 KB Output is correct
8 Correct 75 ms 5740 KB Output is correct
9 Correct 90 ms 6508 KB Output is correct
10 Correct 208 ms 9708 KB Output is correct
11 Correct 74 ms 10220 KB Output is correct
12 Correct 62 ms 8684 KB Output is correct
13 Correct 55 ms 8044 KB Output is correct
14 Correct 38 ms 5484 KB Output is correct
15 Correct 70 ms 10348 KB Output is correct
16 Correct 2 ms 364 KB Output is correct
17 Correct 2 ms 364 KB Output is correct
18 Correct 1 ms 364 KB Output is correct
19 Correct 2 ms 364 KB Output is correct
20 Correct 2 ms 364 KB Output is correct
21 Correct 36 ms 2412 KB Output is correct
22 Correct 28 ms 2048 KB Output is correct
23 Correct 244 ms 9068 KB Output is correct
24 Correct 154 ms 6508 KB Output is correct
25 Correct 131 ms 6380 KB Output is correct
26 Correct 76 ms 4460 KB Output is correct
27 Correct 104 ms 5100 KB Output is correct
28 Correct 176 ms 6636 KB Output is correct
29 Correct 109 ms 4588 KB Output is correct
30 Correct 275 ms 9452 KB Output is correct