Submission #622933

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
622933	2022-08-04T20:33:36 Z	inksamurai	Mag (COCI16_mag)	C++17	24 / 120	677 ms	262144 KB

#include <bits/stdc++.h>
#define int ll
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define vec(...) vector<__VA_ARGS__>
#define _3SgiE60 ios::sync_with_stdio(0),cin.tie(0)
typedef long long ll;
using pii=pair<int,int>;
using vi=vector<int>;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}
// e

const int inf=1e13;

pii compare(const pii&l,const pii&r){
	return l.fi*r.se<r.fi*l.se?l:r;
}

signed main(){
_3SgiE60;
	int n;
	cin>>n;
	vec(vi) adj(n);
	rep(i,n-1){
		int u,v;
		cin>>u>>v;
		u-=1,v-=1;
		adj[u].pb(v);
		adj[v].pb(u);	
	}
	vi a(n);
	rep(i,n){
		cin>>a[i];
	}
	vi dist(n);
	vec(vi) nadj(n);
	auto pre_dfs=[&](auto self,int v,int par,int up)->void{
		if(par!=-1){
			nadj[v].pb(up);
			nadj[up].pb(v);
			if(a[v]>1){
				up=v;
			}else if(a[par]!=1){
				up=v;
			}
		}
		up=v;
		for(auto &u:adj[v]){
			if(u==par) continue;
			dist[u]=dist[v]+1;
			self(self,u,v,up);
		}
	};
	pre_dfs(pre_dfs,0,-1,0);
	adj=nadj;
	pii pns={inf,1};
	vec(vec(pii)) rbts(n);
	auto dfs=[&](auto self,int v,int par)->void{
		vi chs;
		for(auto &u:adj[v]){
			if(u==par) continue;
			chs.pb(u);
			self(self,u,v);
		}
		// if(a[v]>1){
		vec(pii) ps;
		{
			for(auto &u:chs){
				pii p={inf,1};
				rep(i,sz(rbts[u])){
					int _u=rbts[u][i].se;
					int _prod=rbts[u][i].fi;
					int _dist=dist[_u]-dist[v];
					if(i==0){
						p={_prod,_dist};
					}else{
						p=compare(pii{_prod,_dist},p);
					}
				}
				if(!sz(ps)){
					ps.pb(p);
				}else{
					ps.pb(compare(ps.back(),p));
				}
			}
		}
		vec(pii) sp;
		{
			per(j,sz(chs)){
				int u=chs[j];
				pii p={inf,1};
				rep(i,sz(rbts[u])){
					int _u=rbts[u][i].se;
					int _prod=rbts[u][i].fi;
					int _dist=dist[_u]-dist[v];
					if(i==0){
						p={_prod,_dist};
					}else{
						p=compare(pii{_prod,_dist},p);
					}
				}
				if(!sz(sp)){
					sp.pb(p);
				}else{
					sp.pb(compare(sp.back(),p));
				}
			}
			reverse(sp.begin(), sp.end());
		}
		rep(j,sz(chs)){
			int u=chs[j];
			rep(i,sz(rbts[u])){
				int _u=rbts[u][i].se;
				int _prod=rbts[u][i].fi;
				int _dist=dist[_u]-dist[v];
				pii p={inf,1};
				if(j){
					p=compare(p,ps[j-1]);
				}
				if(j+1<sz(chs)){
					p=compare(p,sp[j+1]);
				}
				if(p.fi<inf and p.fi<inf/a[v] and a[v]*_prod<inf/p.fi){
					pii now={p.fi*_prod*a[v],_dist+p.se+1};
					pns=compare(pns,now);
				}
			}
		}
		if(sz(ps)){
			pii p=ps.back();
			if(p.fi<inf/a[v]){
				pns=compare(pns,pii(p.fi*a[v],p.se+1));
			}
		}
		rep(j,sz(chs)){
			int u=chs[j];
			for(auto p:rbts[u]){
				int _u=p.se;
				int _prod=p.fi;
				if(_prod<inf/a[v]){
					rbts[v].pb(pii(_prod*a[v],_u));
				}else{
					// 
				}
			}
		}
		rbts[v].pb(pii(a[v],v));
		pns=compare(pns,pii(a[v],1));
	};
	dfs(dfs,0,-1);
	int g=gcd(pns.fi,pns.se);
	pns.fi/=g;
	pns.se/=g;
	cout<<pns.fi<<"/"<<pns.se<<"\n";
	// print(pns.fi,pns.se);
}

Subtask #1
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	2 ms	640 KB	Output is correct
2	Correct	3 ms	1364 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	3 ms	1364 KB	Output is correct
2	Correct	3 ms	1524 KB	Output is correct

Subtask #3
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	371 ms	262144 KB	Execution killed with signal 9
2	Halted	0 ms	0 KB	-

Subtask #4
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	340 KB	Output is correct
2	Runtime error	447 ms	262144 KB	Execution killed with signal 9
3	Halted	0 ms	0 KB	-

Subtask #5
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	509 ms	262144 KB	Execution killed with signal 9
2	Halted	0 ms	0 KB	-

Subtask #6
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	541 ms	262144 KB	Execution killed with signal 9
2	Halted	0 ms	0 KB	-

Subtask #7
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	677 ms	244516 KB	Output is correct
2	Correct	217 ms	55036 KB	Output is correct
3	Runtime error	459 ms	262144 KB	Execution killed with signal 9
4	Halted	0 ms	0 KB	-

Subtask #8
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	238 ms	61592 KB	Output is correct
2	Runtime error	657 ms	262144 KB	Execution killed with signal 9
3	Halted	0 ms	0 KB	-

Subtask #9
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	550 ms	262144 KB	Execution killed with signal 9
2	Halted	0 ms	0 KB	-

Subtask #10
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Runtime error	550 ms	262144 KB	Execution killed with signal 9
2	Halted	0 ms	0 KB	-

Submission #622933

Compilation message