4302 2023. 03. 22 17:30:13 horvathabel Útadó cpp17 Wrong answer 0/50 777ms 8252 KiB
#include <bits/stdc++.h>
using namespace std;
using ll=long long;
vector<ll> g[50001];
vector<ll>c(50001);
vector<pair<ll,int>> db ; 
ll dfs(int x){
	if (c[x]==0){
		db[x]={1,x};
		return db[x].first;
	} 
	else{
		for (int edge:g[x]){
			db[x].first+=dfs(edge);
		}
		db[x].second=x; 

	}
	return (db[x].first);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int n;
cin>>n;

vector<int> p;
p.resize(n+1);
for (int i=1;i<n;i++)
{
ll x,y;
cin>>x>>y;
db.assign(n+1, {1,0}); 
c[x]++; 
p[y]=x;
g[x].push_back(y);

}
vector<ll> arak(n+1);
for (int i=1; i<n;i++){
	cin>>arak[i-1];
}

dfs(1);
for (int i=1; i<=n;i++) db[i].first=db[i].first*(n-db[i].first);
sort(db.begin(), db.end());
sort(arak.begin(), arak.end());
ll ans=0; 
vector<pair<pair<ll, ll>,ll>> cnt(n-1); 
for (int i=2; i<=n;i++){
	ans+=db[i].first*arak[i-2]*2;
	ans=ans%32609;
	cnt[i-2]={{db[i].second, p[db[i].second]},arak[i-2]};
}
cout<<ans<<endl; 
for (auto x: cnt) cout<<x.first.second<<" "<<x.first.first<<" "<<x.second<<"\n"; 

}
Subtask Sum Test Verdict Time Memory
base 0/50
1 Wrong answer 0/0 4ms 4788 KiB
2 Wrong answer 0/0 456ms 8252 KiB
3 Wrong answer 0/2 4ms 5300 KiB
4 Wrong answer 0/2 4ms 5516 KiB
5 Wrong answer 0/2 4ms 5844 KiB
6 Wrong answer 0/2 4ms 5928 KiB
7 Wrong answer 0/2 4ms 5992 KiB
8 Time limit exceeded 0/8 771ms 5376 KiB
9 Wrong answer 0/2 4ms 6264 KiB
10 Wrong answer 0/2 4ms 6476 KiB
11 Wrong answer 0/2 4ms 6580 KiB
12 Wrong answer 0/2 6ms 6444 KiB
13 Wrong answer 0/2 4ms 6636 KiB
14 Time limit exceeded 0/2 763ms 6012 KiB
15 Time limit exceeded 0/2 741ms 6012 KiB
16 Time limit exceeded 0/2 777ms 6184 KiB
17 Time limit exceeded 0/2 765ms 6420 KiB
18 Time limit exceeded 0/2 758ms 6504 KiB
19 Time limit exceeded 0/2 765ms 6420 KiB
20 Time limit exceeded 0/2 746ms 6516 KiB
21 Time limit exceeded 0/2 767ms 6672 KiB
22 Time limit exceeded 0/2 777ms 6816 KiB
23 Time limit exceeded 0/2 763ms 6840 KiB
24 Time limit exceeded 0/2 763ms 6812 KiB