【hdu5909】Tree Cutting【FWT加速运算】【树形dp】

也许因为在人群中多看了你一眼，再也没能忘掉你容颜

以前总把FFT啊FWT啊当成题目来做，现在要转变思维，把它们当做工具，在某些时候用来加速的工具。

设f[i][j]表示i这个大子树内子树权值为j的子树个数。

那么相当于自己和所有子节点进行异或操作，FWT加速即可。

一开始真没想到，放一个yyb的代码上来，反正长得差不多。

注意

为了保证答案上传，每次要把常数项++，但计算答案的时候也不能算进去，所以要弄回来。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 1111
#define MOD (1000000007)
#define inv2 (500000004)
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int n,m,N;
void FWT(int *P,int opt)
{
    for(int i=2;i<=N;i<<=1)
        for(int p=i>>1,j=0;j<N;j+=i)
            for(int k=j;k<j+p;++k)
            {
                int x=P[k],y=P[k+p];
                P[k]=(x+y)%MOD;P[k+p]=(x-y+MOD)%MOD;
                if(opt==-1)P[k]=1ll*P[k]*inv2%MOD,P[k+p]=1ll*P[k+p]*inv2%MOD; 
            }
}
int f[MAX][1<<10],ans[MAX];
void dfs(int u,int ff)
{
    FWT(f[u],1);
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff)continue;
        dfs(v,u);
        for(int j=0;j<N;++j)f[u][j]=1ll*f[u][j]*f[v][j]%MOD;
    }
    FWT(f[u],-1);f[u][0]=(f[u][0]+1)%MOD;FWT(f[u],1);
}
int main()
{
    int T=read();
    while(T--)
    {
        n=read();m=read();cnt=1;N=m;
        memset(f,0,sizeof(f));memset(h,0,sizeof(h));memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;++i)f[i][read()]++;
        for(int i=1;i<n;++i)
        {
            int u=read(),v=read();
            Add(u,v);Add(v,u);
        }
        dfs(1,0);
        for(int i=1;i<=n;++i)FWT(f[i],-1);
        for(int i=1;i<=n;++i)f[i][0]=(f[i][0]-1+MOD)%MOD;
        for(int i=1;i<=n;++i)
            for(int j=0;j<m;++j)
                ans[j]=(ans[j]+f[i][j])%MOD;
        for(int i=0;i<m;++i,(i==m)?putchar('\n'):putchar(' '))printf("%d",ans[i]);
    }
}