Hdu 6133 启发式合并

今(yi)天(zhi)被NJU(suoyou)的多校虐得满地找牙。。
第一次写启发式合并。。
大概就是大的子树只更新一次，把小的子树往大的里面抽插(233)。。
然后就可以了。。
唯一的问题是我代码能力十分捉急，写出了n个bug。。

//启发式合并get//NJU很牛逼啊//这个sumofsum树状数组也很套路啊//我TM好菜菜啊#include <bits/stdc++.h>using namespace std;typedef long long ll;int a[105000];int id[105000];int b[105000];ll num[105000];ll sum[105000];int lson[105000];int rson[105000];int n;int cmp(int x, int y){    if(a[x]==a[y])return x<y;    return a[x]<a[y];}struct edge{    int to, next;}e[205000];int head[105000];int cnt;int siz[105000];ll ans[105000];ll tmp;ll tot;void init(){    tmp=0;    memset(lson, -1, sizeof(lson));    memset(rson, -1, sizeof(rson));    memset(ans, 0, sizeof(ans));    memset(num, 0, sizeof(num));    memset(sum, 0, sizeof(sum));    memset(head, -1, sizeof(head));    cnt=0;}void add(int u, int v){    e[cnt].to=v;    e[cnt].next=head[u];    head[u]=cnt++;}void dfs(int u, int fa){    siz[u]=1;    for(int i=head[u];~i;i=e[i].next){        if(e[i].to!=fa){            dfs(e[i].to, u);            siz[u]+=siz[e[i].to];            if(rson[u]==-1)rson[u]=e[i].to;            else {                if(siz[e[i].to]>siz[rson[u]]){                    lson[u]=rson[u];                    rson[u]=e[i].to;                }                else {                    lson[u]=e[i].to;                }            }        }    }}int lowbit(int x){    return x&(-x);}void update(ll f[], int pos, int val){    while(pos<=tot){        f[pos]+=val;        pos+=lowbit(pos);    }}ll query(ll f[], int pos){    ll ans=0;    while(pos){        ans+=f[pos];        pos-=lowbit(pos);    }    return ans;}void add(int u){    tmp+=(query(num, tot)-query(num, b[u]))*a[u];    tmp+=query(sum, b[u])+a[u];    update(num, b[u], 1);    update(sum, b[u], a[u]);}void del(int u){    tmp-=(query(num, tot)-query(num, b[u]))*a[u];    tmp-=query(sum, b[u]);    update(num, b[u], -1);    update(sum, b[u], -a[u]);}void addtree(int u){    add(u);    if(lson[u]!=-1)addtree(lson[u]);    if(rson[u]!=-1)addtree(rson[u]);}void deltree(int u){    del(u);    if(lson[u]!=-1)deltree(lson[u]);    if(rson[u]!=-1)deltree(rson[u]);}void dfs1(int u){    if(lson[u]==-1){        if(rson[u]==-1){            add(u);            ans[u]=tmp;            return ;        }        dfs1(rson[u]);        add(u);        ans[u]=tmp;        return;    }    dfs1(lson[u]);    deltree(lson[u]);    dfs1(rson[u]);    addtree(lson[u]);    add(u);    ans[u]=tmp;    return ;}int main(){    int T;    scanf("%d", &T);    while(T--){        init();        scanf("%d", &n);        for(int i=1;i<=n;i++){            scanf("%d", &a[i]);            id[i]=i;        }        sort(id+1, id+1+n, cmp);        b[id[1]]=1;        for(int i=2;i<=n;i++){            if(a[id[i]]==a[id[i-1]])b[id[i]]=b[id[i-1]];            else b[id[i]]=b[id[i-1]]+1;        }        tot=b[id[n]];        for(int i=1;i<n;i++){            int u, v;            scanf("%d%d", &u, &v);            add(u, v);            add(v, u);        }        dfs(1, 0);        dfs1(1);        for(int i=1;i<=n;i++)printf("%lld ", ans[i]);        printf("\n");    }}