2024杭电多校第一场 —— 1003 树

前置知识：树上启发式合并 dsu on tree（建议OI-WiKi上学习）

开三个树状数组分别维护前缀和、平方和、数量即可

#include<bits/stdc++.h>
using namespace std;
#define int unsigned long long
#define all(x) x.begin(),x.end()
#define lowbit(x) (x&(-x))
#define no cout<<"No"<<endl
#define yes cout<<"Yes"<<endl
#define endl '\n'
// #define x first
// #define y second
typedef pair<int,int> PII;
const int N=5e5+10;
const int M=1e6+10;
const int mod=998244353;
const int INF=0x3f3f3f3f3f3f3f3f;

int tr1[M+1],tr2[M+1],tr3[M+1];//前缀和，平方和，数量
void add(int u,int val,int tr[]){
    for(int i=u;i<M;i+=lowbit(i)){
        tr[i]+=val;
    }
}
int query(int u,int tr[]){
    int res=0;
    for(int i=u;i;i-=lowbit(i)){
        res+=tr[i];
    }
    return res;
}


int idx,sz[N],son[N],L[N],R[N],rnk[N];
vector<vector<int>> g(N);
void dfs1(int u,int fa){
    sz[u]=1;
    L[u]=++idx; rnk[idx]=u;
    for(auto v:g[u]){
        if(v==fa)continue;
        dfs1(v,u);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]])son[u]=v;
    }
    R[u]=idx;
}

int a[N],ans[N];
void dfs2(int u,int fa,int flag){
    for(auto v:g[u]){
        if(v==fa||v==son[u])continue;
        dfs2(v,u,0);
    }
    if(son[u])dfs2(son[u],u,1);
    for(auto v:g[u]){
        if(v==fa||v==son[u])continue;
        for(int i=L[v];i<=R[v];i++){
            int j=rnk[i];
            ans[u]+=query(a[j]-1,tr3)*a[j]*a[j]-a[j]*query(a[j]-1,tr1)+query(M,tr2)-query(a[j],tr2)-a[j]*(query(M,tr1)-query(a[j],tr1));
            add(a[j],a[j],tr1);
            add(a[j],a[j]*a[j],tr2);
            add(a[j],1,tr3);
        }
    }
    ans[u]+=query(a[u]-1,tr3)*a[u]*a[u]-a[u]*query(a[u]-1,tr1)+query(M,tr2)-query(a[u],tr2)-a[u]*(query(M,tr1)-query(a[u],tr1));
    ans[u]*=2;
    ans[u]+=ans[son[u]];
    add(a[u],a[u],tr1);
    add(a[u],a[u]*a[u],tr2);
    add(a[u],1,tr3);
    if(!flag){
        for(int i=L[u];i<=R[u];++i){
            int j=rnk[i];
            add(a[j],-a[j],tr1);
            add(a[j],-a[j]*a[j],tr2);
            add(a[j],-1,tr3);
        }
    }
}

void solve(){
    int n;cin>>n;
    for(int i=0;i<n-1;i++){
        int u,v;cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    for(int i=1;i<=n;i++)cin>>a[i];
    dfs1(1,1);
    dfs2(1,1,0);

    int res=0;
    for(int i=1;i<=n;i++){
        res^=ans[i];
    }
    cout<<res<<endl;
}

signed main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int _=1;
    while(_--)solve();
    return 0;
}