9957 2024. 03. 21 18:01:46 111 Erőművek cpp17 Hibás válasz 0/100 101ms 47132 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long

signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
#ifdef CB
	freopen("be2.txt","r",stdin);
//	freopen("out.txt","w",stdout);
#endif
	int N,M;
	cin>>N>>M;
	vector<int>v(N+1);
	int su=0;
	for(int i=1;i<=N;i++){
		cin>>v[i];
		su+=v[i];
	}
	vector<vector<int>>g(N+1);
	for(int i=0;i<M;i++){
		int a,b;
		cin>>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	int ans=INT_MAX;
	vector<int>l(N+1,INT_MAX),a(N+1,0);
	vector<int>s(N+1);
	vector<set<int>>z(N+1);
	auto dfs=[&](auto self,int i)->void{
		for(int j:g[i]){
			if(l[j]==l[i]-1){
				continue;
			}
			if(l[j]!=INT_MAX){
				if(l[j]<l[a[i]]){
					a[i]=j;
				}
				continue;
			}
			l[j]=l[i]+1;
			self(self,j);
			if(l[a[j]]<l[a[i]]){
				a[i]=a[j];
			}
			if(l[a[j]]>l[i]){
				z[i].insert(s[j]);
				int s1=su-s[j],s2,s3;
				auto it=z[j].lower_bound(s[j]/2);
				if(it!=z[j].end()){
					s2=*it;
					s3=su-s1-s2;
					ans=min(ans,max({s1,s2,s3})-min({s1,s2,s3}));
				}
				if(it!=z[j].begin()){
					--it;
					s2=*it;
					s3=su-s1-s2;
					ans=min(ans,max({s1,s2,s3})-min({s1,s2,s3}));
				}
				if(!z[j].empty()){
					s2=*s.begin();
					s3=su-s1-s2;
					ans=min(ans,max({s1,s2,s3})-min({s1,s2,s3}));
					s2=*--s.end();
					s3=su-s1-s2;
					ans=min(ans,max({s1,s2,s3})-min({s1,s2,s3}));
				}
			}
			s[i]+=s[j];
			if(z[j].size()>z[i].size()){
				swap(z[i],z[j]);
			}
			z[i].insert(z[j].begin(),z[j].end());
		}
		s[i]+=v[i];
	};
	l[1]=0;
	dfs(dfs,1);
//	for(int i=1;i<=N;i++){
//		cout<<i<<" : "<<l[i]<<" "<<a[i]<<" "<<s[i]<<endl;
//	}
	cout<<ans<<'\n';
	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 0/100
1 Hibás válasz 0/0 3ms 1832 KiB
2 Hibás válasz 0/0 4ms 3352 KiB
3 Hibás válasz 0/5 3ms 2524 KiB
4 Hibás válasz 0/5 3ms 2748 KiB
5 Hibás válasz 0/5 3ms 3208 KiB
6 Hibás válasz 0/5 3ms 3184 KiB
7 Hibás válasz 0/5 6ms 4744 KiB
8 Hibás válasz 0/5 6ms 4552 KiB
9 Hibás válasz 0/5 9ms 6676 KiB
10 Hibás válasz 0/5 8ms 6920 KiB
11 Hibás válasz 0/6 26ms 14552 KiB
12 Hibás válasz 0/6 23ms 14412 KiB
13 Hibás válasz 0/6 43ms 23148 KiB
14 Hibás válasz 0/6 37ms 22656 KiB
15 Hibás válasz 0/6 64ms 32120 KiB
16 Hibás válasz 0/6 52ms 31464 KiB
17 Hibás válasz 0/6 79ms 39536 KiB
18 Hibás válasz 0/6 67ms 38812 KiB
19 Hibás válasz 0/6 101ms 47132 KiB
20 Hibás válasz 0/6 79ms 46396 KiB