#include <cstdio> #include <vector> #include <algorithm> #include <iostream> using namespace std; #define f first #define s second const int MAXN = 313; const long long INF = 1e18 + 3; int r[MAXN]; int odw[MAXN]; int zab[MAXN]; int kol[MAXN]; long long w[MAXN]; long long ans[MAXN]; vector<pair<int, int> > v[MAXN]; vector<pair<int, int> > v2[MAXN]; long long temp; int tm, cnt; void preDFS(int x, int oj) { kol[++tm] = x; for (auto it:v2[x]) { if (it.f != oj) { v[x].push_back(it); preDFS(it.f, x); } } } void DFS(int x) { temp += w[x]; odw[x] = cnt; for (auto it:v[x]) { if (zab[it.s] != cnt) DFS(it.f); } } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int z, n; long long wyn; int x, y, mx, ile, licz; cin >> z; while (z--) { cin >> n; for (int i = 1; i <= n; i++) { r[i] = odw[i] = zab[i] = kol[i] = 0; v2[i].clear(); v[i].clear(); ans[i] = INF; } cnt = tm = 0; for (int i = 1; i <= n; i++) cin >> w[i]; for (int i = 1; i < n; i++) { cin >> x >> y; v2[x].emplace_back(y, i); v2[y].emplace_back(x, i); } preDFS(1, 0); mx = 1 << (n - 1); for (int i = 0; i < mx; i++) { x = i; cnt++; licz = 0; ile = 1; wyn = 0; while (x) { licz++; if (x & 1) { ile++; zab[licz] = cnt; } x >>= 1; } for (int j = 1; j <= n; j++) { if (odw[kol[j]] != cnt) { temp = 0; DFS(kol[j]); wyn += temp * temp; } } ans[ile] = min(ans[ile], wyn); } for (int i = 1; i <= n; i++) cout << ans[i] << ' '; cout << '\n'; } 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 | #include <cstdio> #include <vector> #include <algorithm> #include <iostream> using namespace std; #define f first #define s second const int MAXN = 313; const long long INF = 1e18 + 3; int r[MAXN]; int odw[MAXN]; int zab[MAXN]; int kol[MAXN]; long long w[MAXN]; long long ans[MAXN]; vector<pair<int, int> > v[MAXN]; vector<pair<int, int> > v2[MAXN]; long long temp; int tm, cnt; void preDFS(int x, int oj) { kol[++tm] = x; for (auto it:v2[x]) { if (it.f != oj) { v[x].push_back(it); preDFS(it.f, x); } } } void DFS(int x) { temp += w[x]; odw[x] = cnt; for (auto it:v[x]) { if (zab[it.s] != cnt) DFS(it.f); } } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int z, n; long long wyn; int x, y, mx, ile, licz; cin >> z; while (z--) { cin >> n; for (int i = 1; i <= n; i++) { r[i] = odw[i] = zab[i] = kol[i] = 0; v2[i].clear(); v[i].clear(); ans[i] = INF; } cnt = tm = 0; for (int i = 1; i <= n; i++) cin >> w[i]; for (int i = 1; i < n; i++) { cin >> x >> y; v2[x].emplace_back(y, i); v2[y].emplace_back(x, i); } preDFS(1, 0); mx = 1 << (n - 1); for (int i = 0; i < mx; i++) { x = i; cnt++; licz = 0; ile = 1; wyn = 0; while (x) { licz++; if (x & 1) { ile++; zab[licz] = cnt; } x >>= 1; } for (int j = 1; j <= n; j++) { if (odw[kol[j]] != cnt) { temp = 0; DFS(kol[j]); wyn += temp * temp; } } ans[ile] = min(ans[ile], wyn); } for (int i = 1; i <= n; i++) cout << ans[i] << ' '; cout << '\n'; } return 0; } |