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
|
using namespace std;
const int maxn = 2e5 + 5; typedef long long ll; ll maxv[maxn << 5]; int ls[maxn << 5], rs[maxn << 5], root[maxn << 5]; ll rt[maxn << 5]; int idx; vector<int> G[maxn]; int n; int color[maxn]; ll ans[maxn];
void (int o){ if(maxv[ls[o]] < maxv[rs[o]]){ maxv[o] = maxv[rs[o]]; rt[o] = rt[rs[o]]; } else if(maxv[ls[o]] > maxv[rs[o]]){ maxv[o] = maxv[ls[o]]; rt[o] = rt[ls[o]]; } else{ maxv[o] = maxv[ls[o]]; rt[o] = rt[ls[o]] + rt[rs[o]]; } }
void update(int &o, int l, int r, int x){ if(!o) o = ++idx; if(l == r){ maxv[o]++; rt[o] = x; return; } int mid = l + r >> 1; if(x <= mid) update(ls[o], l, mid, x); else update(rs[o], mid + 1, r, x); pushup(o); }
int merge(int o1, int o2, int l, int r){ if(!o1 || !o2) return o1 + o2; if(l == r){ maxv[o1] += maxv[o2]; rt[o1] = l; return o1; } int mid = l + r >> 1; ls[o1] = merge(ls[o1], ls[o2], l, mid); rs[o1] = merge(rs[o1], rs[o2], mid + 1, r); pushup(o1); return o1; }
void dfs(int u, int f){ for(auto &v : G[u]){ if(v == f) continue; dfs(v, u); merge(root[u], root[v], 1, n); } update(root[u], 1, n, color[u]); ans[u] = rt[root[u]]; }
int main(){ ios::sync_with_stdio(false), cin.tie(0), cout.tie(0); cin >> n; for(int i = 1; i <= n; i++) cin >> color[i], root[i] = ++idx; for(int i = 1; i < n; i++){ int u, v; cin >> u >> v; G[u].emplace_back(v); G[v].emplace_back(u); } dfs(1, 0); for(int i = 1; i <= n; i++) cout << ans[i] << (i == n ? 'n' : ' '); return 0; }
|
近期评论