题意:
给你一棵树,树上的每个点都有个权值,问每个点到下标比他大的路径权值异或和
题解:
数据范围1e5,如果小一点可以预处理前缀暴力做
这题可以计算二进制每一位对答案的贡献,因为权值范围1e6,计算2^0----2^20就好了
计数的时候用cnt[u][0]记录以u为根异或值为0的路径数量,cnt[u][1]记录异或值为1的数量
统计部分有点类似树分治
主要还是拆位了想到了就很简单了
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define PB push_back
#define MP make_pair
#define ll long long
#define MS(a,b) memset(a,b,sizeof(a))
#define LL (rt<<1)
#define RR (rt<<1|1)
#define lson l,mid,LL
#define rson mid+1,r,RR
#define pii pair<int,int>
#define pll pair<ll,ll>
#define lb(x) (x&(-x))
void In(){freopen("in.in","r",stdin);}
void Out(){freopen("out.out","w",stdout);}
const int N=1e5+10;
const int M=3e5+10;
const int Mbit=1e6+10;
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
const double pi=acos(-1);
const double eps=1e-8;
int n;
vector<int>G[N];
int a[N],b[N],cnt[N][2];
ll sum;
void dfs(int u,int fa)
{
cnt[u][0]=cnt[u][1]=0;
cnt[u][b[u]]++;
sum+=b[u];
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(v==fa)continue;
dfs(v,u);
sum+=cnt[u][0]*cnt[v][1]+cnt[u][1]*cnt[v][0];
cnt[u][b[u]]+=cnt[v][0];
cnt[u][b[u]^1]+=cnt[v][1];
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<n;i++){
int u,v;
scanf("%d%d",&u,&v);
G[u].PB(v);
G[v].PB(u);
}
ll ans=0;
for(int i=0;i<=20;i++){
for(int j=1;j<=n;j++)
if(a[j]&(1<<i))
b[j]=1;
else b[j]=0;
sum=0;
dfs(1,-1);
ans+=(sum<<i);
}
printf("%lld\n",ans);
return 0;
}