刷题记录（NC13611 树）（dfs序）

NC13611 树

题目链接

关键点：

1、题目要求x,y颜色相同，那么从x到y路径上的颜色均与x,y相同

2、那么如果按照dfs的顺序来染色，则对于一个孩子结点，染的颜色要么与父亲相同，要么就是用之前从未出现的颜色来染，因为按照dfs顺序来染色，则染到该点时，该点的父亲、祖先、某些兄弟的颜色已经全染了。如果与除了父亲颜色外其他祖先、兄弟的颜色，那么无法保证从该祖先或者兄弟到该点的颜色均相同。因此该点的染色情况只有两种

3、所以如果按照dfs序来染色，那么该树就为一个数组，就可以不用管树的结构，进行动态规划

4、设dp[i][j]表示，染前i个结点，需要j种颜色，则

dp[i][j] += dp[i][j-1]（该点染的颜色与父亲相同） + dp[i-1][j-1] * (k-(j-1))（该点染一种之前从未出现过的颜色，则选择的颜色数为总颜色数(k) - 之前用的颜色数(j-1)）

5、最后将染所有节点，所需要耗费的颜色(1-k)的总方案数相加即为答案

6、答案可能会很大，要开long long

完整代码：

# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, k;
ll dp[400][400];
const int MOD = 1e9+7;
int main()
{
    scanf("%d%d", &n, &k);
    dp[0][0] = 1;
    for (int i=1; i<=n; i++)
    {
        for (int j=1; j<=k; j++)
        {
            dp[i][j] = (dp[i][j] + dp[i-1][j] + (dp[i-1][j-1]*(k-j+1)%MOD))%MOD;
        }
    }
    ll ans = 0;
    for (int i=1; i<=k; i++)
    {
        ans = (ans+dp[n][i])%MOD;
    }
    cout<<ans<<endl;
    
    return 0;
}