HDU 5909 : Tree Cutting（树形DP + FWT优化）

题目大意：有一棵树有n个结点，每个结点有一个权值，定义一棵树的值为所有结点的值的异或和，让你输出[0,m - 1]内，每个值 $i$ 有多少种子树的值等于 $i$。

题解：考虑树形dp，另dp[i][j]表示以$i$为根节点,异或和为$j$的方案数，转移是一个二重循环，复杂度达到 $m^2$

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);
        memset(tmp,0,sizeof tmp);
        for(int j = 0; j < m; j++) {
            for(int k = 0; k < m; k++) {
                if(!dp[u][k] || !dp[v][j]) continue;
                tmp[k ^ j] += dp[u][k] * dp[v][j] % mod;
                tmp[k ^ j] %= mod;
            }
        }
        for(int k = 0; k < m; k++) {
            dp[u][k] += tmp[k];
            dp[u][k] %= mod;
        }
    }
}

最后要把每个点的答案加起来才能得到总答案。由于复杂度太高这样是铁定领一发TLE。

仔细看转移过程，其实每次转移是一个下标为异或运算的卷积，可以用FWT加速至$m\ast logm$，复杂度降为$n\ast m\ast logm$，勉强能过。

更快的解法是点分治，复杂度为$n\ast m\ast logn$，但常数更小，待补

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e3 + 10;
const int mod = 1e9 + 7;
long long dp[maxn][maxn],val[maxn];
int t,n,m;
vector<int> g[maxn];
long long ans[maxn];
long long tmp1[maxn],tmp2[maxn];
long long inv;
long long fpow(long long a, long long b) {
	long long r = 1;
	while(b) {
		if(b & 1)
			r = r * a % mod;
		a = a * a % mod;
		b >>= 1;
	}
	return r;
}
void fwt(long long a[],int len) {
	for(int s = 2; s <= len; s <<= 1) {
		for(int j = 0; j < len; j += s) {
			for(int k = 0; k < s / 2; k++) {
				long long x = a[j + k];
				long long y = a[j + k + s / 2];
				a[j + k] = (x + y) % mod;
				a[j + k + s / 2] = (x - y + mod) % mod;
			}
		}
	}
}
void ufwt(long long a[],int len) {
	for(int s = 2; s <= len; s <<= 1) {
		for(int i = 0; i < len; i += s) {
			for(int j = 0; j < s / 2; j++) {
				long long x = a[i + j],y = a[i + j + s / 2];
				a[i + j] = ((x + y) % mod) * inv % mod;
				a[i + j + s / 2] = ((x - y + mod) % mod) * inv % mod;
			}
		}
	}	
}
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);
		memset(tmp1,0,sizeof tmp1);
		memset(tmp2,0,sizeof tmp2);
		for(int j = 0; j < m; j++) {
			tmp1[j] = dp[u][j];
			tmp2[j] = dp[v][j];
		}
		fwt(tmp1,m);
		fwt(tmp2,m);
		for(int j = 0; j < m; j++)
			tmp1[j] = tmp1[j] * tmp2[j] % mod;
		ufwt(tmp1,m);
		for(int k = 0; k < m; k++) {
			dp[u][k] += tmp1[k];
			dp[u][k] %= mod;
		}
	}
}
int main() {
	scanf("%d",&t);
	inv = fpow(2,mod - 2);
	while(t--) {
		scanf("%d%d",&n,&m);
		memset(dp,0,sizeof dp);
		memset(ans,0,sizeof ans);
		for(int i = 1; i <= n; i++) {
			scanf("%lld",&val[i]);
			g[i].clear();
		}
		for(int i = 1; i <= n - 1; 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++) {
			for(int j = 1; j <= n; j++) {
				ans[i] += dp[j][i];
				ans[i] %= mod;
			}
			if(i) printf(" ");
			printf("%lld",ans[i]);
		}
		puts("");
	}
	return 0;
}