#include <bits/stdc++.h>
using namespace std;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef long long LL;
typedef pair<int,int> PI;
typedef pair<LL,LL> PLL;
typedef unsigned long long ULL;
typedef pair<double,double> PD;
#define FOR(x, b, e) for(int x = b; x<= (e); x++)
#define FORD(x, b, e) for(int x = b; x>= (e); x--)
#define REP(x, n) for(int x = 0; x<(n); ++x)
#define ALL(c) (c).begin(), (c).end()
#define SIZE(x) ((int)(x).size())
#define PB push_back
#define IN insert
#define ST first
#define ND second
#define INF 2000000011
#define MOD 1000000007
#define MAXS 310
VI graf[MAXS];
LL val[MAXS];
int P[MAXS];
LL koszty[MAXS];
void makeTree(int v){
for(auto x:graf[v]){
if(P[x]==-1){
P[x]=v;
makeTree(x);
}
}
}
LL param;
LL tKoszt;
LL dfs(int v){
// cout<<v<<" i "<<val[v]<<"ech\n";
LL sum=val[v];
for(auto x:graf[v]){
if(P[x]==v)
sum+=dfs(x);
}
if((1<<v)¶m){
tKoszt+=sum*sum;
sum=0;
}
return sum;
}
void go(){
int n;
cin>>n;
REP(i,n){
P[i]=-1;
cin>>val[i];
koszty[i]=LONG_LONG_MAX;
graf[i].clear();
}
REP(i,n-1){
int a,b;
cin>>a>>b;
a--;b--;
graf[a].PB(b);
graf[b].PB(a);
}
P[0]=0;
makeTree(0);
REP(i,1<<n){//{bit maska przedstawia czy krawedz wierzcholka i do ojca jest zablokowana}
param=i;
tKoszt=0;
LL v1=dfs(0);
tKoszt+=v1*v1;
koszty[__builtin_popcount(i)]=min(koszty[__builtin_popcount(i)],tKoszt);
}
REP(i,n)
cout<<koszty[i]<<" ";
cout<<"\n";
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
int t;
cin>>t;
while(t--)
go();
}
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 | #include <bits/stdc++.h> using namespace std; typedef vector<int> VI; typedef vector<VI> VVI; typedef long long LL; typedef pair<int,int> PI; typedef pair<LL,LL> PLL; typedef unsigned long long ULL; typedef pair<double,double> PD; #define FOR(x, b, e) for(int x = b; x<= (e); x++) #define FORD(x, b, e) for(int x = b; x>= (e); x--) #define REP(x, n) for(int x = 0; x<(n); ++x) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define PB push_back #define IN insert #define ST first #define ND second #define INF 2000000011 #define MOD 1000000007 #define MAXS 310 VI graf[MAXS]; LL val[MAXS]; int P[MAXS]; LL koszty[MAXS]; void makeTree(int v){ for(auto x:graf[v]){ if(P[x]==-1){ P[x]=v; makeTree(x); } } } LL param; LL tKoszt; LL dfs(int v){ // cout<<v<<" i "<<val[v]<<"ech\n"; LL sum=val[v]; for(auto x:graf[v]){ if(P[x]==v) sum+=dfs(x); } if((1<<v)¶m){ tKoszt+=sum*sum; sum=0; } return sum; } void go(){ int n; cin>>n; REP(i,n){ P[i]=-1; cin>>val[i]; koszty[i]=LONG_LONG_MAX; graf[i].clear(); } REP(i,n-1){ int a,b; cin>>a>>b; a--;b--; graf[a].PB(b); graf[b].PB(a); } P[0]=0; makeTree(0); REP(i,1<<n){//{bit maska przedstawia czy krawedz wierzcholka i do ojca jest zablokowana} param=i; tKoszt=0; LL v1=dfs(0); tKoszt+=v1*v1; koszty[__builtin_popcount(i)]=min(koszty[__builtin_popcount(i)],tKoszt); } REP(i,n) cout<<koszty[i]<<" "; cout<<"\n"; } int main(){ ios::sync_with_stdio(0); cin.tie(0); int t; cin>>t; while(t--) go(); } |
English