#pragma GCC optimize ("O3")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> ii;
typedef vector<int> vi;
const int INF = 0x3f3f3f3f;
#define FOR(i,b,e) for(int i = (b); i < (e); i++)
#define TRAV(x,a) for(auto &x: (a))
#define SZ(x) ((int)x.size())
#define PB push_back
#define X first
#define Y second
const int N = 305;
int n;
int val[N], rozm[N];
vector<pair<ll, ll>> dp[N][N];
vi G[N];
ll obl(pair<ll, ll> &v){
return v.X + v.Y*v.Y;
}
void dfs(int v, int p){
dp[v][0] = {{0, val[v]}};
rozm[v] = 1;
TRAV(u, G[v]){
if(u == p) continue;
dfs(u, v);
rozm[v] += rozm[u];
}
TRAV(u, G[v]){
if(u == p) continue;
for(int k = rozm[v]-1; k >= 0; k--){
vector<pair<ll, ll>> nowe;
FOR(i, 0, k+1){
TRAV(el1, dp[v][i]){
TRAV(el2, dp[u][k-i]){
nowe.PB({el1.X+el2.X, el1.Y+el2.Y});
}
if(k-i-1 < 0) continue;
TRAV(el2, dp[u][k-i-1]){
nowe.PB({el1.X+el2.X+el2.Y*el2.Y, el1.Y});
}
}
}
sort(nowe.begin(), nowe.end());
dp[v][k].clear();
TRAV(x, nowe){
if(SZ(dp[v][k]) == 0 || x.Y < dp[v][k].back().Y) dp[v][k].PB(x);
}
nowe.clear();
ll mink = 1ll*INF*INF;
for(int i = SZ(dp[v][k])-1; i >= 0; i--){
if(obl(dp[v][k][i]) < mink) nowe.PB(dp[v][k][i]);
mink = min(mink, obl(dp[v][k][i]));
}
dp[v][k] = nowe;
}
}
}
ll wart(int k){
ll ret = 1ll*INF*INF;
TRAV(x, dp[0][k]) ret = min(ret, obl(x));
return ret;
}
void solve(){
cin >> n;
FOR(i, 0, n) cin >> val[i];
FOR(i, 1, n){
int a, b;
cin >> a >> b;
a--; b--;
G[a].PB(b);
G[b].PB(a);
}
dfs(0, 0);
FOR(i, 0, n) cout << wart(i) << ' ';
cout << '\n';
FOR(i, 0, n) G[i].clear();
FOR(i, 0, n) FOR(k, 0, rozm[i]) dp[i][k].clear();
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
int tt; cin >> tt;
FOR(te, 0, tt)
solve();
return 0;
}
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 | #pragma GCC optimize ("O3") #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int, int> ii; typedef vector<int> vi; const int INF = 0x3f3f3f3f; #define FOR(i,b,e) for(int i = (b); i < (e); i++) #define TRAV(x,a) for(auto &x: (a)) #define SZ(x) ((int)x.size()) #define PB push_back #define X first #define Y second const int N = 305; int n; int val[N], rozm[N]; vector<pair<ll, ll>> dp[N][N]; vi G[N]; ll obl(pair<ll, ll> &v){ return v.X + v.Y*v.Y; } void dfs(int v, int p){ dp[v][0] = {{0, val[v]}}; rozm[v] = 1; TRAV(u, G[v]){ if(u == p) continue; dfs(u, v); rozm[v] += rozm[u]; } TRAV(u, G[v]){ if(u == p) continue; for(int k = rozm[v]-1; k >= 0; k--){ vector<pair<ll, ll>> nowe; FOR(i, 0, k+1){ TRAV(el1, dp[v][i]){ TRAV(el2, dp[u][k-i]){ nowe.PB({el1.X+el2.X, el1.Y+el2.Y}); } if(k-i-1 < 0) continue; TRAV(el2, dp[u][k-i-1]){ nowe.PB({el1.X+el2.X+el2.Y*el2.Y, el1.Y}); } } } sort(nowe.begin(), nowe.end()); dp[v][k].clear(); TRAV(x, nowe){ if(SZ(dp[v][k]) == 0 || x.Y < dp[v][k].back().Y) dp[v][k].PB(x); } nowe.clear(); ll mink = 1ll*INF*INF; for(int i = SZ(dp[v][k])-1; i >= 0; i--){ if(obl(dp[v][k][i]) < mink) nowe.PB(dp[v][k][i]); mink = min(mink, obl(dp[v][k][i])); } dp[v][k] = nowe; } } } ll wart(int k){ ll ret = 1ll*INF*INF; TRAV(x, dp[0][k]) ret = min(ret, obl(x)); return ret; } void solve(){ cin >> n; FOR(i, 0, n) cin >> val[i]; FOR(i, 1, n){ int a, b; cin >> a >> b; a--; b--; G[a].PB(b); G[b].PB(a); } dfs(0, 0); FOR(i, 0, n) cout << wart(i) << ' '; cout << '\n'; FOR(i, 0, n) G[i].clear(); FOR(i, 0, n) FOR(k, 0, rozm[i]) dp[i][k].clear(); } int main(){ ios::sync_with_stdio(0); cin.tie(0); int tt; cin >> tt; FOR(te, 0, tt) solve(); return 0; } |
English