#include <bits/stdc++.h>
using namespace std;
const long long INF=(1ll<<60);
int n, ed[310][2];
long long a[310];
vector <int> g[310];
int odw[310], nr, licz;
long long wyn[310];
long long dfs(int w)
{
long long ans=a[w];
odw[w]=nr;
for(auto xd : g[w])
{
if(odw[xd]!=nr)
{
ans+=dfs(xd);
}
}
return ans;
}
void solve()
{
scanf("%d", &n);
for(int i=1; i<=n; ++i)
{
scanf("%lld", &a[i]);
wyn[i]=INF;
}
for(int i=0; i<n-1; ++i)
{
scanf("%d%d", &ed[i][0], &ed[i][1]);
}
long long cur, h;
for(int i=(1<<(n-1))-1; i>=0; --i)
{
for(int j=1; j<=n; ++j)
{
g[j].clear();
}
int x=i;
licz=n;
for(int j=0; j<n-1; ++j)
{
if(x&1)
{
g[ed[j][0]].push_back(ed[j][1]);
g[ed[j][1]].push_back(ed[j][0]);
--licz;
}
x>>=1;
}
cur=0;
++nr;
for(int j=1; j<=n; ++j)
{
if(odw[j]!=nr)
{
h=dfs(j);
cur+=h*h;
}
}
wyn[licz]=min(wyn[licz], cur);
}
for(int i=1; i<=n; ++i)
{
printf("%lld ", wyn[i]);
}
printf("\n");
}
int main()
{
int t;
scanf("%d", &t);
for(int i=0; i<t; ++i)
{
solve();
}
}
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 | #include <bits/stdc++.h> using namespace std; const long long INF=(1ll<<60); int n, ed[310][2]; long long a[310]; vector <int> g[310]; int odw[310], nr, licz; long long wyn[310]; long long dfs(int w) { long long ans=a[w]; odw[w]=nr; for(auto xd : g[w]) { if(odw[xd]!=nr) { ans+=dfs(xd); } } return ans; } void solve() { scanf("%d", &n); for(int i=1; i<=n; ++i) { scanf("%lld", &a[i]); wyn[i]=INF; } for(int i=0; i<n-1; ++i) { scanf("%d%d", &ed[i][0], &ed[i][1]); } long long cur, h; for(int i=(1<<(n-1))-1; i>=0; --i) { for(int j=1; j<=n; ++j) { g[j].clear(); } int x=i; licz=n; for(int j=0; j<n-1; ++j) { if(x&1) { g[ed[j][0]].push_back(ed[j][1]); g[ed[j][1]].push_back(ed[j][0]); --licz; } x>>=1; } cur=0; ++nr; for(int j=1; j<=n; ++j) { if(odw[j]!=nr) { h=dfs(j); cur+=h*h; } } wyn[licz]=min(wyn[licz], cur); } for(int i=1; i<=n; ++i) { printf("%lld ", wyn[i]); } printf("\n"); } int main() { int t; scanf("%d", &t); for(int i=0; i<t; ++i) { solve(); } } |
English