//#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <typename T>
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
//~ while (clock()<=69*CLOCKS_PER_SEC)
#define ll long long
#define ld long double
#define pi pair<int,int>
#define pd pair<ld,ld>
#define ft first
#define sd second
#define pb push_back
#define eb emplace_back
#define FOR(i,a,b) for(int i=(a); i<=(b);i++)
#define F(i,a,b) FOR(i,(a),(b)-1)
#define REV(i,a,b) for(int i=(a); i>=(b);i--)
#define VI vector<int>
#define VPI vector<pi>
#define VPD vector<pd>
#define PI 3.14159265
#define all(x) (x).begin(), (x).end()
#define sz(a) (int)((a).size())
#define int long long
template<class TH> void _dbg(const char *sdbg, TH h){cerr<<sdbg<<"="<<h<<"\n";}
template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) {
while(*sdbg!=',')cerr<<*sdbg++;cerr<<"="<<h<<","; _dbg(sdbg+1, a...);
}
#define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__)
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
const long long K = 1000;
const long long KK = K*K;
const long long MIL = KK*K;
const long long INF = MIL*MIL;
const long long MOD = 1e9 + 7;
const long long N = 30, M=310;
int par[N];
int sum[N];
int Find(int x) {
return par[x] == x ? par[x] : par[x] = Find(par[x]);
}
void Union(int a, int b) {
a = Find(a);
b = Find(b);
sum[a] += sum[b];
par[b] = a;
}
void solve() {
int n;
cin >> n;
VI A(n + 1), res(n + 1, INF);
FOR(i, 1, n)
cin >> A[i];
VPI G;
F(i, 0, n - 1) {
int a, b;
cin >> a >> b;
G.eb(a, b);
}
F(i, 0, 1 << n) {
bitset<30> bits = i;
FOR(i, 1, n) {
par[i] = i;
sum[i] = A[i];
}
F(j, 0, n - 1) {
if (!bits[j]) {
int a = G[j].ft, b = G[j].sd;
if (Find(a) != Find(b))
Union(a, b);
}
}
int tmp = 0;
FOR(j, 1, n) {
if (par[j] == j) {
tmp += sum[j] * sum[j];
}
}
res[bits.count()] = min(res[bits.count()], tmp);
}
FOR(i, 0, n - 1) {
cout << res[i] << " ";
}
cout << "\n";
}
int32_t main() {
// freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
ios_base::sync_with_stdio(0);
cin.tie(0);
// cout.tie(0);
cerr.tie(0);
cout << setprecision(9) << fixed;
cerr << setprecision(6) << fixed;
int test = 1, f;
cin >> test;
F(_test, 0, test) {
//cout<<"Case #"<<_test + 1<<": ";
solve();
// if(_test == 1)
// return 0;
}
}
/*
5
2 -2 1 -2 4
*/
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 119 120 121 122 123 124 125 | //#pragma GCC optimize("Ofast") #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; using namespace __gnu_pbds; template <typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; //~ while (clock()<=69*CLOCKS_PER_SEC) #define ll long long #define ld long double #define pi pair<int,int> #define pd pair<ld,ld> #define ft first #define sd second #define pb push_back #define eb emplace_back #define FOR(i,a,b) for(int i=(a); i<=(b);i++) #define F(i,a,b) FOR(i,(a),(b)-1) #define REV(i,a,b) for(int i=(a); i>=(b);i--) #define VI vector<int> #define VPI vector<pi> #define VPD vector<pd> #define PI 3.14159265 #define all(x) (x).begin(), (x).end() #define sz(a) (int)((a).size()) #define int long long template<class TH> void _dbg(const char *sdbg, TH h){cerr<<sdbg<<"="<<h<<"\n";} template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) { while(*sdbg!=',')cerr<<*sdbg++;cerr<<"="<<h<<","; _dbg(sdbg+1, a...); } #define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__) mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()); const long long K = 1000; const long long KK = K*K; const long long MIL = KK*K; const long long INF = MIL*MIL; const long long MOD = 1e9 + 7; const long long N = 30, M=310; int par[N]; int sum[N]; int Find(int x) { return par[x] == x ? par[x] : par[x] = Find(par[x]); } void Union(int a, int b) { a = Find(a); b = Find(b); sum[a] += sum[b]; par[b] = a; } void solve() { int n; cin >> n; VI A(n + 1), res(n + 1, INF); FOR(i, 1, n) cin >> A[i]; VPI G; F(i, 0, n - 1) { int a, b; cin >> a >> b; G.eb(a, b); } F(i, 0, 1 << n) { bitset<30> bits = i; FOR(i, 1, n) { par[i] = i; sum[i] = A[i]; } F(j, 0, n - 1) { if (!bits[j]) { int a = G[j].ft, b = G[j].sd; if (Find(a) != Find(b)) Union(a, b); } } int tmp = 0; FOR(j, 1, n) { if (par[j] == j) { tmp += sum[j] * sum[j]; } } res[bits.count()] = min(res[bits.count()], tmp); } FOR(i, 0, n - 1) { cout << res[i] << " "; } cout << "\n"; } int32_t main() { // freopen("input.txt", "r", stdin); // freopen("output.txt", "w", stdout); ios_base::sync_with_stdio(0); cin.tie(0); // cout.tie(0); cerr.tie(0); cout << setprecision(9) << fixed; cerr << setprecision(6) << fixed; int test = 1, f; cin >> test; F(_test, 0, test) { //cout<<"Case #"<<_test + 1<<": "; solve(); // if(_test == 1) // return 0; } } /* 5 2 -2 1 -2 4 */ |
English