【Codeforces】600E。Lomsat gelral

Codeforces 600E。Lomsat gelral

木の上のDSUについて学びます

なぜ木にDSUのDFSではないでしょうか？？？

この質問は、この木は、軽鎖および重鎖を分割入れ、その後、削除与えた彼の息子の息子を点灯すべき光の寄与を終え最初の光の息子、対処されます

この点の寄与を計算するために、私は、光の寄与の全て息子を再度追加しました

この点は、光の息子、無人権であるならば、私は再びそれを空にするツリー全体を横断します

実際に、私は彼らができない一つ明確な1、配列全体が空であると思いますので、現在のサブツリートラバーサル再び空に

まあ、私は良い短いと感じ、本当にこれは......

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define eps 1e-10
#define MAXN 100005
#define ba 47
//#define ivorysi
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
    if(c == '-') f = -1;
    c = getchar();
    }
    while(c >= '0' && c <= '9') {
    res = res * 10 +c - '0';
    c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
    out(x / 10);
    }
    putchar('0' + x % 10);
}
struct node {
    int to,next;
}E[MAXN * 2];
int head[MAXN],sumE;
int N,c[MAXN],siz[MAXN],son[MAXN];
int val[MAXN],mval;
int64 sum,ans[MAXN];
bool vis[MAXN];
void add(int u,int v) {
    E[++sumE].to = v;
    E[sumE].next = head[u];
    head[u] = sumE;
}
void dfs(int u,int fa) {
    siz[u] = 1;
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa) {
        dfs(v,u);
        siz[u] += siz[v];
        if(siz[v] > siz[son[u]]) son[u] = v;
    }
    }
}
void change(int u,int fa,int k) {
    val[c[u]] += k;
    if(k == -1) vis[u] = 0;
    if(val[c[u]] >= mval) {
    if(val[c[u]] > mval) {mval = val[c[u]];sum = 0;}
    sum += c[u];
    }
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa && (!vis[v] || k == -1)) {
        change(v,u,k);
    }
    }
}
void calc(int u,int fa,bool f) {
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa && v != son[u]) {
        calc(v,u,0);
    }
    }
    if(son[u]) {calc(son[u],u,1);vis[son[u]] = 1;}
    change(u,fa,1);
    ans[u] = sum;
    if(!f) {change(u,fa,-1);sum = 0;mval = 0;}
}
void Solve() {
    read(N);
    for(int i = 1 ; i <= N ; ++i) read(c[i]);
    int a,b;
    for(int i = 1 ; i < N ; ++i) {
    read(a);read(b);
    add(a,b);add(b,a);
    }
    dfs(1,0);
    calc(1,0,1);
    for(int i = 1 ; i <= N ; ++i) {
    out(ans[i]);space;
    }
    enter;
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
}