第五届新疆省ACM-ICPC程序设计竞赛（重现赛） J.异或的路径

最新推荐文章于 2019-07-03 09:48:19 发布

fffefw

最新推荐文章于 2019-07-03 09:48:19 发布

阅读量285

点赞数

分类专栏：树形dp

本文链接：https://blog.csdn.net/qq_40642465/article/details/90742556

版权

树形dp 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这是一道树形dp题，定义dp[i]表示根到i节点路径的异或和。然后画几组数据可以找到规律，答案是 $ans = \sum_{i=1}^n\sum_{j=1}^ndp[i]\oplus dp[j]$ ，dp[i]可以dfs求出来，但是这个求和就不好处理了，我当时写的时候误以为异或运算与加法运算满足分配律，把dp[i]提了出来，结果gg了。后来看了大佬的代码才学会了怎么处理这个式子，开一个cnt[]数组,cnt[i]表示第i位为1的dp[j]的个数，,然后对于这个Sigma dp[i]^dp[j] 就可以按位算贡献了，对于假如dp[i]第j位的值为1，那就找这个dp数组里第j位为0的数的个数n-cnt[j]，然后统计第j位对值的贡献，同理对于dp[i]第j位值为0就找有几个1，这样就可以以31*n的复杂度求解了

#include<bits/stdc++.h>
using namespace std;
#define fuck(x) cout<<#x<<"     "<<x<<endl;
#define ll long long
const int maxn=1e5+10;
const int mod=1e9+7;
struct node
{
    int v,w;
    node(int v=0,int w=0)
    {
        this->v=v;
        this->w=w;
    }
};
int dp[maxn],cnt[32];
ll ans;
vector<node>g[maxn];
void dfs(int now,int fa)
{
    for(int i=1;i<=31;i++)
        cnt[i]+=(dp[now]&(1<<(i-1)))?1:0;
    int sz=g[now].size();
    for(int i=0;i<sz;i++)
    {
        int v=g[now][i].v,w=g[now][i].w;
        if(v==fa) continue;
        dp[v]=dp[now]^w;
        dfs(v,now);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,w;
        scanf("%d%d",&u,&w);
        g[i+1].push_back(node(u,w));
        g[u].push_back(node(i+1,w));
    }
    dfs(1,-1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=31;j++)
            if(dp[i]&(1<<(j-1)))
                ans+=1LL*(n-cnt[j])*((1<<(j-1))%mod)%mod,ans%=mod;
            else
                ans+=1LL*cnt[j]*((1<<(j-1))%mod)%mod,ans%=mod;
    printf("%lld\n",ans);
    return 0;
}