hdu5909 Tree Cutting - FWT 加速集合异或（*）

http://acm.hdu.edu.cn/showproblem.PHP?pid=5909

问题描述

Byteasar有一棵$n$个点的无根树，节点依次编号为$1$到$n$，其中节点$i$的权值为v_iv​i​​。

定义一棵树的价值为它所有点的权值的异或和。

现在对于每个$[0,m)$的整数$k$，请统计有多少$T$的非空连通子树的价值等于$k$。

一棵树$T$的连通子树就是它的一个连通子图，并且这个图也是一棵树。

一个暴力的做法就是 树dp

[html]  view plain  copy

1. void dfs(int x,int fa)  
2. {  
3.     dp[x][a[x]]=1; //与自己异或之后的值为a【x】的方案数  
4.     for (int i=0; i<mp[x].size(); i++)  
5.     {  
6.         int v=mp[x][i];  
7.         if(v==fa) continue;  
8.         dfs(v,x);  
9.         for (int i=0; i<m; i++)  
10.             tmp[i]=dp[x][i];  
11.         solve(dp[x],dp[v],m);//其中这里求得其实是 当前dp[x]的所有值与dp[v]的所有值 异或的结果, **  
12.         for (int i=0; i<m; i++)  
13.             dp[x][i]=(dp[x][i]+tmp[i])%mod;  
14.     }  
15.     for (int i=0;i<m;i++)  
16.         ans[i]=(ans[i]+dp[x][i])%mod;  
17. }  

**处 的solve，暴力两个for 是n^2的，如果用fwt加速，可以做到nlogn

复杂度从n^3变为n*n*logn

恩，就是这样咯

[cpp]  view plain  copy

1. #include<bits/stdc++.h>  
2. using namespace std;  
3. const int N=1e3+100,mod=1e9+7,rev=(mod+1)>>1;  
4. int a[N],dp[N][N],ans[N];//dp[u][i],表示u为根的数，xor值得到i的方案数  
5. vector <int >mp[N];  
6. int n,m;  
7. int tmp[N];  
8. void FWT(int *a,int n)  
9. {  
10.     for(int d=1; d<n; d<<=1)  
11.         for(int m=d<<1,i=0; i<n; i+=m)  
12.             for(int j=0; j<d; j++)  
13.             {  
14.                 int x=a[i+j],y=a[i+j+d];  
15.                 a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;   
16.             }  
17. }  
18.   
19. void UFWT(int *a,int n)  
20. {  
21.     for(int d=1; d<n; d<<=1)  
22.         for(int m=d<<1,i=0; i<n; i+=m)  
23.             for(int j=0; j<d; j++)  
24.             {  
25.                 int x=a[i+j],y=a[i+j+d];  
26.                 a[i+j]=1LL*(x+y)*rev%mod,a[i+j+d]=(1LL*(x-y)*rev%mod+mod)%mod;   
27.             }  
28. }  
29.   
30. void solve(int *a,int *b,int n)  
31. {  
32.     FWT(a,n);  
33.     FWT(b,n);  
34.     for(int i=0; i<n; i++)   a[i]=1LL*a[i]*b[i]%mod;  
35.     UFWT(a,n);  
36. }  
37. void dfs(int x,int fa)  
38. {  
39.     dp[x][a[x]]=1; //与自己异或之hou的值为0的方案数  
40.     for (int i=0; i<mp[x].size(); i++)  
41.     {  
42.         int v=mp[x][i];  
43.         if(v==fa) continue;  
44.         dfs(v,x);  
45.         for (int i=0; i<m; i++)  
46.             tmp[i]=dp[x][i];  
47.         solve(dp[x],dp[v],m);  
48.         for (int i=0; i<m; i++)  
49.             dp[x][i]=(dp[x][i]+tmp[i])%mod;  
50.     }  
51.     for (int i=0;i<m;i++)  
52.         ans[i]=(ans[i]+dp[x][i])%mod;  
53. }  
54.   
55.   
56. int main()  
57. {  
58.     int t,u,v;  
59.     cin>>t;  
60.     while(t--)  
61.     {  
62.         cin>>n>>m;  
63.         memset(dp,0,sizeof dp);  
64.         memset(ans,0,sizeof ans);  
65.         for (int i=1; i<=n; i++) mp[i].clear();  
66.         for (int i=1; i<=n; i++)  scanf("%d",&a[i]);  
67.         for (int i=1; i<n; i++)  
68.         {  
69.             scanf("%d%d",&u,&v);  
70.             mp[u].push_back(v);  
71.             mp[v].push_back(u);  
72.         }  
73.         dfs(1,0);  
74.         for (int i=0;i<m;i++)  
75.             printf("%d%c",ans[i],i==m-1?'\n':' ');  
76.   
77.     }  
78.   
79.   
80.   
81.     return 0;  
82. }