#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
#define ST first
#define ND second
#define PB push_back
#define SIZE(a) ((int)a.size())
template<class T, class U>
ostream& operator<<(ostream &stream, pair<T, U> &p) {
stream << "(" << p.ST << "," << p.ND << ")";
return stream;
}
template<class T>
ostream& operator<<(ostream &stream, vector<T> &v) {
stream << "[";
for(auto elem : v) {
stream << elem << ", ";
}
stream << "]";
return stream;
}
typedef vector<vector<pll>> vvpll;
const int maxn = 310;
vector<int> v[maxn];
int a[maxn];
vvpll delete_dominated(vvpll &sub) {
vvpll res;
double sub_count=0, res_count=0;
for(int i=SIZE(sub)-1; i >= 0; i--) {
vector<pll> &v = sub[i];
sort(v.begin(), v.end(), [](pll &a, pll &b){return a > b;});
vector<pll> u, z;
for(pll x : v) {
while(!u.empty() && u.back().ST >= x.ST && u.back().ND >= x.ND) {
u.pop_back();
}
u.PB(x);
}
ll minv = u.back().ST*u.back().ST+u.back().ND;
for(int j=SIZE(u)-1; j >= 0; j--) {
ll val = u[j].ST*u[j].ST+u[j].ND;
if(val <= minv) {
z.PB(u[j]);
}
}
res.PB(z);
sub_count+=SIZE(v);
res_count+=SIZE(z);
}
reverse(res.begin(), res.end());
// cerr << sub_count << "/" << res_count << "=" << sub_count/res_count << "\n";
return res;
}
vvpll dfs(int q, int p) {
vvpll res(1);
res[0].PB({ll(a[q]), 0});
for(int b : v[q]) {
if(b == p) {
continue;
}
vvpll sub = dfs(b, q);
vvpll tmp(SIZE(sub)+SIZE(res)-1);
for(int i=0; i < SIZE(res); i++) {
for(int j=0; j < SIZE(sub); j++) {
for(pll x : res[i]) {
for(pll y : sub[j]) {
tmp[i+j].PB({x.ST+y.ST, x.ND+y.ND});
}
}
}
}
res = delete_dominated(tmp);
}
res.PB({});
for(int i=SIZE(res)-2; i >= 0; i--) {
for(pll x : res[i]) {
res[i+1].PB({0, x.ST*x.ST+x.ND});
}
}
// cerr << q << ": " << SIZE(res) << "\n";
// cerr << res << "\n\n";
return delete_dominated(res);
}
int main() {
ios_base::sync_with_stdio(0);
int t;
cin >> t;
while(t--) {
int n;
cin >> n;
for(int i=1; i <= n; i++) {
cin >> a[i];
v[i].clear();
}
for(int i=0; i < n-1; i++) {
int b, c;
cin >> b >> c;
v[b].PB(c);
v[c].PB(b);
}
auto res = dfs(1, -1);
for(int i=1; i <= n; i++) {
cout << res[i][0].ND << " ";
}
cout << "\n";
}
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | #include<bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int, int> pii; typedef pair<ll, ll> pll; #define ST first #define ND second #define PB push_back #define SIZE(a) ((int)a.size()) template<class T, class U> ostream& operator<<(ostream &stream, pair<T, U> &p) { stream << "(" << p.ST << "," << p.ND << ")"; return stream; } template<class T> ostream& operator<<(ostream &stream, vector<T> &v) { stream << "["; for(auto elem : v) { stream << elem << ", "; } stream << "]"; return stream; } typedef vector<vector<pll>> vvpll; const int maxn = 310; vector<int> v[maxn]; int a[maxn]; vvpll delete_dominated(vvpll &sub) { vvpll res; double sub_count=0, res_count=0; for(int i=SIZE(sub)-1; i >= 0; i--) { vector<pll> &v = sub[i]; sort(v.begin(), v.end(), [](pll &a, pll &b){return a > b;}); vector<pll> u, z; for(pll x : v) { while(!u.empty() && u.back().ST >= x.ST && u.back().ND >= x.ND) { u.pop_back(); } u.PB(x); } ll minv = u.back().ST*u.back().ST+u.back().ND; for(int j=SIZE(u)-1; j >= 0; j--) { ll val = u[j].ST*u[j].ST+u[j].ND; if(val <= minv) { z.PB(u[j]); } } res.PB(z); sub_count+=SIZE(v); res_count+=SIZE(z); } reverse(res.begin(), res.end()); // cerr << sub_count << "/" << res_count << "=" << sub_count/res_count << "\n"; return res; } vvpll dfs(int q, int p) { vvpll res(1); res[0].PB({ll(a[q]), 0}); for(int b : v[q]) { if(b == p) { continue; } vvpll sub = dfs(b, q); vvpll tmp(SIZE(sub)+SIZE(res)-1); for(int i=0; i < SIZE(res); i++) { for(int j=0; j < SIZE(sub); j++) { for(pll x : res[i]) { for(pll y : sub[j]) { tmp[i+j].PB({x.ST+y.ST, x.ND+y.ND}); } } } } res = delete_dominated(tmp); } res.PB({}); for(int i=SIZE(res)-2; i >= 0; i--) { for(pll x : res[i]) { res[i+1].PB({0, x.ST*x.ST+x.ND}); } } // cerr << q << ": " << SIZE(res) << "\n"; // cerr << res << "\n\n"; return delete_dominated(res); } int main() { ios_base::sync_with_stdio(0); int t; cin >> t; while(t--) { int n; cin >> n; for(int i=1; i <= n; i++) { cin >> a[i]; v[i].clear(); } for(int i=0; i < n-1; i++) { int b, c; cin >> b >> c; v[b].PB(c); v[c].PB(b); } auto res = dfs(1, -1); for(int i=1; i <= n; i++) { cout << res[i][0].ND << " "; } cout << "\n"; } } |
English