43012023-03-22 15:02:56horvathabelÚtadócpp17Időlimit túllépés 20/50800ms8976 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;
for (int i=1; i<n;i++){
	ll x;
	cin>>x; 
	arak.push_back(x);
}

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; 
for (int i=2; i<=n;i++){
	ans+=db[i].first*arak[i-2]*2;
	ans=ans%32609;
	cnt.push_back({{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"; 

}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base20/50
1Elfogadva0/04ms4784 KiB
2Elfogadva0/0456ms8976 KiB
3Elfogadva2/24ms5188 KiB
4Elfogadva2/24ms5492 KiB
5Elfogadva2/24ms5612 KiB
6Elfogadva2/24ms5916 KiB
7Elfogadva2/24ms6128 KiB
8Időlimit túllépés0/8800ms5624 KiB
9Elfogadva2/26ms6656 KiB
10Elfogadva2/26ms6912 KiB
11Elfogadva2/24ms6892 KiB
12Elfogadva2/26ms7212 KiB
13Elfogadva2/24ms7100 KiB
14Időlimit túllépés0/2750ms6052 KiB
15Időlimit túllépés0/2759ms5944 KiB
16Időlimit túllépés0/2757ms5980 KiB
17Időlimit túllépés0/2764ms5968 KiB
18Időlimit túllépés0/2751ms6156 KiB
19Időlimit túllépés0/2762ms6288 KiB
20Időlimit túllépés0/2763ms6400 KiB
21Időlimit túllépés0/2777ms6408 KiB
22Időlimit túllépés0/2758ms6636 KiB
23Időlimit túllépés0/2768ms6864 KiB
24Időlimit túllépés0/2773ms6784 KiB