CF102832F Strange Memory

题面传送门
CCPC也出水题。
我们考虑这种东西怎么做，首先肯定枚举lca，然后变成了统计一个点在子树内的点对的两两lca是当前节点的权值异或和。
如果这个是方案数那么就很好做我们直接dsu on tree统计一下即可。但是这个异或让我们很不好做。
对于这种东西我们考虑拆位。
然后对于每一位分别计算答案即可，时间复杂度\(O(nlog^2n)\)
code:

#include<bits/stdc++.h>
#pragma GCC optimize(2)
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define ll long long
#define db double
#define N 100000
#define W (1<<20)
#define mod 1000000007
#define eps (1e-7)
#define U unsigned int
#define it iterator
#define Gc() getchar() 
#define Me(x,y) memset(x,y,sizeof(x))
using namespace std;
int n,A[N+5],F[W+5][17][2],siz[N+5],son[N+5],st[N+5<<1],sh,x,y;ll Ans;
struct yyy{int to,z;};
struct ljb{int head,h[N+5];yyy f[N+5<<1];I void add(int x,int y){f[++head]=(yyy){y,h[x]};h[x]=head;}}s;
I void Get(int x){for(re int i=0;i<17;i++) F[A[x]][i][(x>>i)&1]++;st[++sh]=A[x];}
I void Query(int x,int id){for(re int i=0;i<17;i++) Ans+=(1ll<<i)*F[A[id]^A[x]][i][!((x>>i)&1)];}
I void Make(int x,int last){yyy tmp;siz[x]=1;for(int i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^last&&(Make(tmp.to,x),siz[x]+=siz[tmp.to],siz[son[x]]<siz[tmp.to]&&(son[x]=tmp.to));}
I void insert(int x,int last){yyy tmp;re int i;Get(x);for(i=s.h[x];i;i=tmp.z)tmp=s.f[i],tmp.to^last&&(insert(tmp.to,x),0);}
I void find(int x,int last,int id){yyy tmp;re int i;Query(x,id);for(i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^last&&(find(tmp.to,x,id),0);}
I void dfs(int x,int last){
	re int i,lasttop;yyy tmp;for(i=s.h[x];i;i=tmp.z){tmp=s.f[i];if(tmp.to^last&&tmp.to^son[x]){lasttop=sh;dfs(tmp.to,x);while(lasttop^sh) Me(F[st[sh--]],0);}}
	son[x]&&(dfs(son[x],x),0);for(i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^last&&tmp.to^son[x]&&(find(tmp.to,x,x),insert(tmp.to,x),0);Get(x);
}
int main(){
//  	freopen("1.in","r",stdin);freopen("1.out","w",stdout); 
	re int i;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",&A[i]);for(i=1;i<n;i++) scanf("%d%d",&x,&y),s.add(x,y),s.add(y,x);Make(1,0);dfs(1,0);printf("%lld\n",Ans);
}