牛客21758B：树

牛客21758B：树

题目描述

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。

import java.util.*;
public class Main{
    public static void main(String[] args) {
        int mod = 1000000007;
        //注意：k种颜色不一定都要用
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();//节点
        int k = sc.nextInt();//颜色
        long[][] dfs = new long[n+1][k+1];//留出一个用于计数，更好处理边界
        dfs[0][0]=1;//设置初始值 dp[1][1]时只有K种方案
        long res = 0 ;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= k ; j++) {
                //dp算法 找规律发现状态转移方程，也可以参考leetcode 图像渲染
                //dp[i][j]=dp[i-1][j-1]*(k-(j-1))+dp[i-1][j];
                //假设k=4 n=8
                //dp[8][3]是不是就等于dp[7][3],而dp[8][4]是不是就等于dp[7][3](还可以染的颜色种数)..
                //那么就是dp[8][4]=dp[7][3](k-3)+还有一种就是dp[7][4]上个7个格子染色4个继续下一个
                dfs[i][j] = (dfs[i-1][j]+dfs[i-1][j-1]*(k-j+1))%mod;
            }
        }
        //然后最后一行的值加起来，既是总方案数
        for (int i = 1; i <= k; i++) {
            res = (res + dfs[n][i])%mod;
        }
        System.out.println(res);
    }
}