Wannafly模拟赛 树【思维+Dp】

题目描述

shy有一颗树，树有n个结点。有k种不同颜色的染料给树染色。一个染色方案是合法的，当且仅当对于所有相同颜色的点对(x,y)，x到y的路径上的所有点的颜色都要与x和y相同。请统计方案数。

输入描述:

第一行两个整数n，k代表点数和颜色数；
接下来n-1行，每行两个整数x,y表示x与y之间存在一条边；

输出描述:

输出一个整数表示方案数（mod 1e9+7）。

示例1

输入

4 3
1 2
2 3
2 4

输出

39

备注:

对于30%的数据，n≤10, k≤3；
对于100%的数据，n,k≤300。

不难想到，问题和树的形状是无关的。

因为我们每种颜色染色的树上的部分一定是一个联通块，那么我们染成的若干个联通块拆分出来，其实就是若干个子树。

那么过程我们Dp一下就能统计结果了。

设定Dp【i】【j】表示i个点，用了j种颜色的方案数。

那么有：Dp【i】【j】=Dp【i-1】【j】+Dp【i-1】【j-1】*（k-(j-1)）；

Ac代码：

#include<stdio.h>
#include<string.h>
using namespace std;
#define ll long long int
ll dp[600][600];
const ll mod=1000000000+7;
int main()
{
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=1;i<=n-1;i++)
        {
            int tmp;
            scanf("%d",&tmp);
            scanf("%d",&tmp);
        }
        ll ans=0;
        dp[0][0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                dp[i][j]=dp[i-1][j]+dp[i-1][j-1]*(k-(j-1));
                dp[i][j]%=mod;
                if(i==n)ans+=dp[i][j];
            }
        }
        printf("%lld\n",ans);
    }
}