JSCPC I Neko and tree 树形dp+前缀和优化

题意是说给你一个树，然后上面有一些关键结点，统计多少种选择点集的方案，使得这个集合里面，任意两个结点的距离都小于等于k。
我们考虑$f[i][j]$表示i这个子树中距离i的最大距离为j的总方案数。转移的时候通过枚举子树转移，对每一个新加入的子树的方案都要乘上原先的方案数，并且做完之后把现在的单独的贡献加进去。$f[u][max(i,j+1)]+=f[u][i]\ast f[v][j]$，然后这个单步转移是$O(k^2)$的。这里对每次转移都处理一次前缀和，单步转移可降为$O(k)$。
若根结点是点集中的结点，那么本身可选可不选，那么要将此处的贡献翻倍并且加上只选自身的特殊的情形。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=5e3+7;
const int mod=1e9+7;
vector<int> G[N];  
int n,m,k;  
ll dp[N][N],sum[N][N],f[N],ans;
bool tag[N];
void cal(int u,int v) {
	for(int i=1;i<k;i++) f[i]=dp[u][i]*sum[v][min(i-1,k-i-1)]%mod;
	for(int i=1;i<k;i++) f[i]=(f[i]+sum[u][min(i,k-i)]*dp[v][i-1])%mod;
	for(int i=1;i<=k/2;i++) f[i]=(f[i]+mod-dp[u][i]*dp[v][i-1]%mod)%mod;
	for(int i=1;i<=k;i++) {
		dp[u][i]=(dp[u][i]+f[i]+dp[v][i-1])%mod;
		sum[u][i]=(sum[u][i-1]+dp[u][i])%mod;
	}
}
void dfs(int u,int fa) {
	for(auto &v:G[u]) {
		if(v==fa) continue;
		dfs(v,u);
		cal(u,v);
	}
	if(tag[u]) {
		sum[u][0]=dp[u][0]=1;
		for(int i=1;i<=k;i++) {
			dp[u][i]=(2LL*dp[u][i])%mod;
			sum[u][i]=(sum[u][i-1]+dp[u][i])%mod;
		}
	}
}
int main() {
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<n;i++) {
		int u,v;
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	for(int i=1;i<=m;i++) {
		int x;
		scanf("%d",&x);
		tag[x]=1;
	}
	dfs(1,0);
	ll ans=sum[1][k];
	for(int i=2;i<=n;i++) 
		ans=(ans+dp[i][k]);
	printf("%lld\n",ans); 
	return 0;
}