HDU 5909 Tree Cutting（树形DP+FWT）

Description

定义一棵树的权值为所有点权值的异或和，给出一棵$n$个节点的树和每个点的权值，问该树有多少颗非空子树其权值为$k$，其中$k\in [0,m)$

Input

第一行一个整数$T$表示用例组数，每组用例首先输入两个整数$n,m$，之后输入$n$个整数$val_i$表示$i$点的权值，最后$n-1$行每行两个整数$u,v$表示一条树边，保证$m$必然可以写成$2^k$形式$(1\le T\le 10,1\le n\le 1000,1\le m\le 2^{10},0\le val_i<m)$

Output

输出$m$个整数分别表示有权值为$0$~$m-1$的非空子树个数

Sample Input

2
4 4
2 0 1 3
1 2
1 3
1 4
4 4
0 1 3 1
1 2
1 3
1 4

Sample Output

3 3 2 3
2 4 2 3

Solution

设$dp[u][i]$表示以$u$为根的子树权值为$i$的方案数，假设$u$的儿子节点为$v_1,v_2,...,v_k$，假设已经考虑了前$j-1$个节点，现在考虑是否加入第$j$个儿子节点，则有转移$dp[u][i]+=\sum\limits dp[u][s]\cdot dp[v_j][i$^$s]$，用$FWT$加速转移即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1111;
#define mod 1000000007
#define inv2 500000004
int T,n,m,val[maxn],dp[maxn][maxn],ans[maxn];
vector<int>g[maxn];
void FWT(int a[],int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
                {  
                    int x=a[i+j],y=a[i+j+d];  
                    a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod; 
                } 
}   
void UFWT(int a[],int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
            {  
                int x=a[i+j],y=a[i+j+d];  
                a[i+j]=(ll)inv2*(x+y)%mod,a[i+j+d]=(ll)inv2*(x-y+mod)%mod;   
            }  
} 
int a[maxn];
void dfs(int u,int fa)
{
    dp[u][val[u]]=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa)continue;
        dfs(v,u);
        for(int j=0;j<m;j++)a[j]=dp[u][j];
        FWT(a,m);FWT(dp[v],m);
        for(int j=0;j<m;j++)a[j]=(ll)a[j]*dp[v][j]%mod;
        UFWT(a,m);
        for(int j=0;j<m;j++)dp[u][j]=(dp[u][j]+a[j])%mod;
    }
    for(int i=0;i<m;i++)ans[i]=(ans[i]+dp[u][i])%mod;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)g[i].clear();
        memset(dp,0,sizeof(dp));
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++)scanf("%d",&val[i]);
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v),g[v].push_back(u);
        }
        dfs(1,0);
        for(int i=0;i<m;i++)printf("%d%c",ans[i],i==m-1?'\n':' ');
    }
    return 0;
}