【Codeforces 766E】 Mahmoud and a xor trip 【二进制计数】

题意：

给你一棵树,树上的每个点都有个权值,问每个点到下标比他大的路径权值异或和

题解：

数据范围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;
}