fwt优化+树形DP HDU 5909

 //fwt优化+树形DP HDU 5909
 //见官方题解
 // BestCoder Round #88 http://bestcoder.hdu.edu.cn/

 #include <bits/stdc++.h>
 // #include <iostream>
 // #include <cstdio>
 // #include <cstdlib>
 // #include <algorithm>
 // #include <vector>
 // #include <queue>
 // #include <math.h>
 using namespace std;
 #define LL long long
 typedef pair<int,int> pii;
 const int inf = 0x3f3f3f3f;
 const int MOD =1e9+;
 const int N =1e3+;
 #define clc(a,b) memset(a,b,sizeof(a))
 const double eps = 1e-;
 void fre() {freopen("in.txt","r",stdin);}
 void freout() {freopen("out.txt","w",stdout);}
 inline int read() {int x=,f=;char ch=getchar();while(ch>''||ch<'') {if(ch=='-') f=-; ch=getchar();}while(ch>=''&&ch<='') {x=x*+ch-'';ch=getchar();}return x*f;}
 const int rev = (MOD+)>>;
 int a[N],tem[N];
 int n,m;
 vector<int> g[N];
 int dp[N][N];
 int ans[N];
 void FWT(int *a,int n)  {
     for(int d=; d<n; d<<=)
         for(int m=d<<,i=; i<n; i+=m)
             for(int j=; 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=; d<n; d<<=)
         for(int m=d<<,i=; i<n; i+=m)
             for(int j=; j<d; j++) {
                 int x=a[i+j],y=a[i+j+d];
                 a[i+j]=1LL*(x+y)*rev%MOD,a[i+j+d]=(1LL*(x-y)*rev%MOD+MOD)%MOD;
             }
 }  

 void solve(int *a,int *b,int n)  {
     FWT(a,n);
     FWT(b,n);
     for(int i=; i<n; i++)   a[i]=1LL*a[i]*b[i]%MOD;
     UFWT(a,n);
 }  

 void dfs(int u,int f){
     dp[u][a[u]]=;
     for(int i=;i<(int)g[u].size();i++){
         int v=g[u][i];
         if(v==f) continue;
         dfs(v,u);
         for(int j=;j<m;j++) tem[j]=dp[u][j];
         solve(dp[u],dp[v],m);
         for(int j=;j<m;j++) dp[u][j]=(dp[u][j]+tem[j])%MOD;
     }
     for(int i=;i<m;i++) ans[i]=(ans[i]+dp[u][i])%MOD;
 }
 int main(){
     // fre();
     int T;
     scanf("%d",&T);
     while(T--){
         clc(dp,);
         clc(ans,);
         scanf("%d%d",&n,&m);
         for(int i=;i<=n;i++) {
             scanf("%d",&a[i]);
             g[i].clear();
         }
         for(int i=;i<=n-;i++){
             int u,v;
             scanf("%d%d",&u,&v);
             g[u].push_back(v);
             g[v].push_back(u);
         }
         dfs(,);
         for(int i=;i<m;i++){
             printf("%d%c",ans[i],i==m- ? '\n':' ');
         }
     }
     return ;
 }