9960 2024. 03. 21 19:08:53 111 Erőművek cpp17 Időlimit túllépés 30/100 287ms 25232 KiB
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

template<typename T>
using ordered_set = tree<T, null_type, less_equal<T>, rb_tree_tag, tree_order_statistics_node_update>;

#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;
	int ans2=0;
	vector<int>l(N+1,INT_MAX),a(N+1,0);
	vector<int>s(N+1);
	vector<ordered_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);
			s[i]+=s[j];
			if(l[a[j]]<l[a[i]]){
				a[i]=a[j];
			}
			if(l[a[j]]>l[i]){
				z[j].insert(s[j]);
				int k=su-s[j];
				for(int kk:z[j]){
					if(ans>max({k,kk,su-k-kk})-min({k,kk,su-k-kk})){
						ans=max({k,kk,su-k-kk})-min({k,kk,su-k-kk});
						ans2=0;
					}
					if(ans==max({k,kk,su-k-kk})-min({k,kk,su-k-kk})){
						ans2++;
					}
				}
			}
			if(z[j].size()>z[i].size()){
				swap(z[i],z[j]);
			}
			if(l[a[j]]>l[i])
			for(int k:z[j]){
				for(int kk:z[i]){
					if(ans>max({k,kk,su-k-kk})-min({k,kk,su-k-kk})){
						ans=max({k,kk,su-k-kk})-min({k,kk,su-k-kk});
						ans2=0;
					}
					if(ans==max({k,kk,su-k-kk})-min({k,kk,su-k-kk})){
						ans2++;
					}
				}
			}
			for(int k:z[j]){
				z[i].insert(k);
			}
		}
		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';
	cout<<ans2<<'\n';
	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 30/100
1 Elfogadva 0/0 3ms 1888 KiB
2 Elfogadva 0/0 12ms 4028 KiB
3 Elfogadva 5/5 4ms 2788 KiB
4 Elfogadva 5/5 4ms 2808 KiB
5 Elfogadva 5/5 8ms 3556 KiB
6 Elfogadva 5/5 8ms 3560 KiB
7 Elfogadva 5/5 83ms 6916 KiB
8 Elfogadva 5/5 97ms 6440 KiB
9 Időlimit túllépés 0/5 250ms 6736 KiB
10 Időlimit túllépés 0/5 263ms 5300 KiB
11 Időlimit túllépés 0/6 287ms 11524 KiB
12 Időlimit túllépés 0/6 279ms 9472 KiB
13 Időlimit túllépés 0/6 259ms 14976 KiB
14 Időlimit túllépés 0/6 263ms 13068 KiB
15 Időlimit túllépés 0/6 270ms 18344 KiB
16 Időlimit túllépés 0/6 243ms 16688 KiB
17 Időlimit túllépés 0/6 263ms 22020 KiB
18 Időlimit túllépés 0/6 272ms 19912 KiB
19 Időlimit túllépés 0/6 280ms 25232 KiB
20 Időlimit túllépés 0/6 263ms 22328 KiB