不知道怎么称呼的题目：树形DP计数

题意：给出1棵树（n<=300），以及有k（<=300）种颜色。现在要给每个节点上色，满足：任意两个相同颜色的点x，y，其简单路径上的点必须也是这个颜色。求方案总数。

题解：首先，染色规则其实说明了一件事情：相同颜色的点是一个块。那么我们只需要求出这棵树用1/2/3/……/k种颜色染色的方案数，然后分别乘上A(k,1)/A(k,2)/A(k,3)/……/A(k,k)就行了。那么比较明显这就是一个树形DP了。用dp[ node ][ kk ]表示node以及node以下的点，用kk种颜色染色的方案数，其中有：父节点的颜色号不小于子节点的颜色号。（因为组合数是最后乘上的，所以这一步只需要每种方案中，每个点有自己的标号就可以，至于怎么组合最后在考虑）

设node为temp的父亲 转移方程式： dp ' [ i+j ] += dp[ node ][ i ]*dp[ temp ][ j ] （temp和node点的颜色不同，这样总共就有i+j种颜色了。dp[ node ][ i ]统计的每一种方案而言，算上dp[ temp ][ j ]的每种方案，都会构成一个唯一的新方案，至于标号不需要考虑，只要颜色方案确定了，标号必然确定。）dp ‘ [ i+j-1 ] += dp[ node ][ i ]*dp[ temp ][ j ]（temp和node颜色相同。）然后再把dp ’ 数组重新赋给dp[ node ]。因为当多了一个孩子的时候，原来dp[ node ]统计的方案数量 由于没有考虑新孩子，都变得不合法了，新算的这个dp ' 才是合法的。

初始条件是dp[ i ][ 1 ] =1意思是 i 这个点染上一个“某种颜色”的方案数=1，这个“某种颜色”在具体计算dp[ i ]的时候，会根据情况得到相应的意义。比如叶子节点dp[ i ][ 1 ]=1表示用1种颜色染掉 i 及其孩子（不存在的）有一种方案。i 的 爹爹是 j 。dp[ j ][ 1 ] =1，根据转移方程：i+j那种情况表示“某种颜色”不同于 i 的。i+j-1那种情况下表示 “某种颜色”和 i 的相同。。。。。。啰里啰唆啰里啰唆。。。

注意：牛客网上边的LL占位符不是I64d

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAX = 305;
const LL MOD = 1e9+7;
vector<int> E[MAX];
LL dp[MAX][MAX];
LL f[MAX];
LL C[MAX][MAX];
LL A[MAX];

int n,k;
int u,v;
void input(){
	scanf("%d%d",&n,&k);
	for (int i=1;i<n;i++){
		scanf("%d%d",&v,&u);
		E[v].push_back(u);
		E[u].push_back(v);
	}
	memset(dp,0,sizeof dp);
}
void dfs(int node,int father){
	if (E[node].size()==1&&node!=1){
		dp[node][1] = 1;
		return;
	}
	dp[node][1]=1;
	for (int temp:E[node]){
		if (temp==father){
			continue;
		}
		dfs(temp,node);
		memset(f,0,sizeof f);
		for (int i=1;i<=k;i++){
			for (int j=1;j<=k;j++){
			if (i+j<=k){
				f[i+j]+=dp[temp][j]*dp[node][i];
				f[i+j]%=MOD;
			}
			if (i+j-1<=k&&i+j-1>0)
			f[i+j-1]+=dp[temp][j]*dp[node][i];
			f[i+j-1]%=MOD;
			}
		}
		for (int i=1;i<=k;i++){
			dp[node][i] = f[i];
		}
	}
}
void print(){
	for (int i=1;i<=n;i++){
		for (int j=1;j<=k;j++){
			printf("dp[%d][%d]=%d\n",i,j,dp[i][j]);
		}
	}
}
void solve(){
	dfs(1,0);
	LL ans =0;
	for (int i=1;i<=k;i++){
		ans+=(1LL*dp[1][i]*C[k][i]%MOD*A[i]%MOD);
		ans%=MOD;
	}
//	print();
	cout<<ans<<endl;
}
void init(){
	C[1][0]=1;
	C[1][1]=1;
	for (int i=2;i<=300;i++){
		C[i][0]=1;
		C[i][i]=1;
		for (int j=1;j<i;j++){
			C[i][j] = C[i-1][j-1]+C[i-1][j];
			C[i][j]%=MOD;
		}
	}
	A[1]=1;
	A[0]=1;
	for (int i=2;i<=300;i++){
		A[i] = A[i-1]*i;
		A[i]%=MOD;
	}
}
int main(){
	init();
	input();
	solve();
	return 0;
}

，还是用cout吧。为什么不开LL让自己WA一次啊。。。为什么中间没有膜又让自己WA一次啊。。

Code：