HDU 6540 树形DP

http://acm.hdu.edu.cn/showproblem.php?pid=6540

题意：给你一颗树和一些关键点，关键点的子集中，任意两点的距离不大于k的子集个数

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int a[5005];
vector<int> mp[5005];
const int mod = 1000000007;
int vis[5005];
int f[5005][5005];
int dep[5005];
int tmp[5005];
void add(int &x, int y)
{
	x = (x+y)%mod;
}
void dfs(int now,int pre)
{
	for(auto chi:mp[now])	
	{
		if(chi==pre)
			continue;
		dfs(chi,now);
		dep[now] = max(dep[now],dep[chi]+1); 
		memcpy(tmp, f[now], sizeof(tmp));
		for(int i=0;i<=dep[now]&&i<=k;i++)
			for(int j=0;j<=dep[chi]&&i<=k;j++)
				if(i+j+1<=k)
					add(tmp[max(i,j+1)],1LL*f[now][i]*f[chi][j]%mod);
		for(int i=0;i<=dep[chi];i++)
			add(tmp[i+1],f[chi][i]);
		for(int i=0;i<=dep[now];i++)
			f[now][i] = tmp[i];
	}
	if(vis[now])
	{
		f[now][0] = 1;
		for(int i=1;i<=dep[now]&&i<=k;i++)
		{
			add(f[now][i],f[now][i]);
		}
	}
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k;
	int x,y;
	for(int i=0;i<n-1;i++)
	{
		cin>>x>>y;
		mp[x].push_back(y);
		mp[y].push_back(x);
	}
	for(int i=0;i<m;i++)
	{
		cin>>x;
		vis[x] = 1;
	}
	dfs(1,0);
	int ans = 0;
	for(int i=0;i<=dep[1];i++)
		ans = (ans+f[1][i])%mod;
	cout<<ans<<endl;
	return 0;
}