解题报告：HDU_5909:Tree Cutting 树上FWT

题目链接

题意：

给定一棵树，树上每个结点有它的权值，现在问在树上选一些连通的点，使得这些点的权值的异或结果 等于0~m-1的时对应的方案数。

思路：

对每个子树统计选定根的方案数，那么答案为：

因为每个分支可以选也可以不选，那么在进行FWT之前对每个儿子进行dp[son][0]++的操作

代码：

#include<bits/stdc++.h>

const int mod = 1e9+7;
const int N = 1050;
using namespace std;

void FWT(long long a[],int l,int on){
   for(int d=1;d<l;d<<=1){
      for(int m=d<<1,i=0;i<l;i+=m){
         for(int j=0;j<d;j++){
            long long x = a[i+j],y=a[i+j+d];
            if( (a[i+j]=x+y) >= mod ) a[i+j]-=mod;
            if( (a[i+j+d]=x-y)<0  ) a[i+j+d]+=mod;
            if(on<0){
               long long ni = 500000004LL;
               a[i+j] = a[i+j] * ni % mod;
               a[i+j+d] = a[i+j+d] * ni % mod;
            }
         }
      }
   }
}


int n,m;
int val[N];
vector<int>G[N];
long long dp[N][N];

void dfs(int x,int fa=0){
   dp[x][val[x]]++;
   for(int i=0;i<G[x].size();i++){
      int t = G[x][i];
      if(t!=fa){
         FWT(dp[x],m,1);
         dfs(t,x);
         dp[t][0]++;
         FWT(dp[t],m,1);
         for(int i=0;i<m;i++){
            dp[x][i] = dp[x][i] * dp[t][i] % mod;
         }FWT(dp[x],m,-1);
      }
   }for(int i=0;i<m;i++){
      if((dp[0][i]+=dp[x][i])>=mod)dp[0][i]-=mod;
   }
   return ;
}

int main()
{
   int T;
   scanf("%d",&T);
   while(T--){
      scanf("%d%d",&n,&m);
      memset(dp,0,sizeof(dp));
      for(int i=1;i<=n;i++){
         scanf("%d",&val[i]);
         G[i].clear();
      }for(int i=1,a,b;i<n;i++){
         scanf("%d%d",&a,&b);
         G[a].emplace_back(b);
         G[b].emplace_back(a);
      }dfs(1);
      for(int i=0;i<m;i++){
         printf("%I64d%c",dp[0][i],i<m-1?' ':'\n');
      }
   }return 0;
}