#include <bits/stdc++.h> using namespace std; void __print(int x) { cerr << x; } void __print(int64_t x) { cerr << x; } void __print(uint32_t x) { cerr << x; } void __print(uint64_t x) { cerr << x; } void __print(float x) { cerr << x; } void __print(double x) { cerr << x; } void __print(long double x) { cerr << x; } void __print(char x) { cerr << '\'' << x << '\''; } void __print(const char *x) { cerr << '\"' << x << '\"'; } void __print(const string &x) { cerr << '\"' << x << '\"'; } void __print(bool x) { cerr << (x ? "true" : "false"); } template<typename T, typename V> void __print(const pair<T, V> &x) { cerr << '('; __print(x.first); cerr << ", "; __print(x.second); cerr << ')'; } template<typename T> void __print(const T &x) { int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}"; } void _print() { cerr << "]\n"; } template <typename T, typename... V> void _print(T t, V... v) { __print(t); if (sizeof...(v)) cerr << ", "; _print(v...); } #ifndef ONLINE_JUDGE #define debug(x...) cerr << " > [" << #x << "] = ["; _print(x) #else #define debug(x...) #endif struct Edge { int to, id; Edge(int _to, int _id) : to(_to), id(_id) {} }; const int64_t oo = 1e15; inline int pop(int x) { return __builtin_popcount(x); } int64_t dfs(int u, int mask, vector<bool>& vis, const vector<vector<Edge>>& g, const vector<int64_t>& w) { vis[u] = true; int64_t subgraph_sum = 0; for(auto v : g[u]) { if(vis[v.to]) continue; if(mask & (1 << v.id)) continue; subgraph_sum += dfs(v.to, mask, vis, g, w); } return subgraph_sum + w[u]; } int64_t sum(int mask, const vector<vector<Edge>>& g, const vector<int64_t>& w) { vector<bool> vis(w.size(), false); int64_t ret = 0; for(int i = 0; i < (int)w.size(); i++) { if(vis[i]) continue; int64_t cur = dfs(i, mask, vis, g, w); ret += cur * cur; } return ret; } void solve() { int n; cin >> n; vector<vector<Edge>> g(n); vector<int64_t> w(n); vector<int64_t> res(n, oo); for(int i = 0; i < n; i++) { cin >> w[i]; } for(int i = 0; i < n - 1; i++) { int u, v; cin >> u >> v; u--; v--; g[u].emplace_back(v, i); g[v].emplace_back(u, i); } int len = (1 << (n - 1)); for(int mask = 0; mask < len; mask++) { res[pop(mask)] = min(res[pop(mask)], sum(mask, g, w)); } for(int64_t num : res) cout << num << " "; cout << "\n"; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int t; cin >> t; while(t--) 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 | #include <bits/stdc++.h> using namespace std; void __print(int x) { cerr << x; } void __print(int64_t x) { cerr << x; } void __print(uint32_t x) { cerr << x; } void __print(uint64_t x) { cerr << x; } void __print(float x) { cerr << x; } void __print(double x) { cerr << x; } void __print(long double x) { cerr << x; } void __print(char x) { cerr << '\'' << x << '\''; } void __print(const char *x) { cerr << '\"' << x << '\"'; } void __print(const string &x) { cerr << '\"' << x << '\"'; } void __print(bool x) { cerr << (x ? "true" : "false"); } template<typename T, typename V> void __print(const pair<T, V> &x) { cerr << '('; __print(x.first); cerr << ", "; __print(x.second); cerr << ')'; } template<typename T> void __print(const T &x) { int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}"; } void _print() { cerr << "]\n"; } template <typename T, typename... V> void _print(T t, V... v) { __print(t); if (sizeof...(v)) cerr << ", "; _print(v...); } #ifndef ONLINE_JUDGE #define debug(x...) cerr << " > [" << #x << "] = ["; _print(x) #else #define debug(x...) #endif struct Edge { int to, id; Edge(int _to, int _id) : to(_to), id(_id) {} }; const int64_t oo = 1e15; inline int pop(int x) { return __builtin_popcount(x); } int64_t dfs(int u, int mask, vector<bool>& vis, const vector<vector<Edge>>& g, const vector<int64_t>& w) { vis[u] = true; int64_t subgraph_sum = 0; for(auto v : g[u]) { if(vis[v.to]) continue; if(mask & (1 << v.id)) continue; subgraph_sum += dfs(v.to, mask, vis, g, w); } return subgraph_sum + w[u]; } int64_t sum(int mask, const vector<vector<Edge>>& g, const vector<int64_t>& w) { vector<bool> vis(w.size(), false); int64_t ret = 0; for(int i = 0; i < (int)w.size(); i++) { if(vis[i]) continue; int64_t cur = dfs(i, mask, vis, g, w); ret += cur * cur; } return ret; } void solve() { int n; cin >> n; vector<vector<Edge>> g(n); vector<int64_t> w(n); vector<int64_t> res(n, oo); for(int i = 0; i < n; i++) { cin >> w[i]; } for(int i = 0; i < n - 1; i++) { int u, v; cin >> u >> v; u--; v--; g[u].emplace_back(v, i); g[v].emplace_back(u, i); } int len = (1 << (n - 1)); for(int mask = 0; mask < len; mask++) { res[pop(mask)] = min(res[pop(mask)], sum(mask, g, w)); } for(int64_t num : res) cout << num << " "; cout << "\n"; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int t; cin >> t; while(t--) solve(); return 0; } |