牛客练习赛-树（染色dp）

最新推荐文章于 2023-03-04 12:48:59 发布

伯纳乌纯白

最新推荐文章于 2023-03-04 12:48:59 发布

阅读量370

点赞数

CC 4.0 BY-SA版权

分类专栏：算法数据结构 C/C++ 文章标签：动态规划算法 c++

本文链接：https://blog.csdn.net/weixin_44522586/article/details/124633979

本文介绍了如何解决一个关于树染色的问题，其中染色方案必须满足路径上相同颜色节点的条件。利用动态规划的方法，通过状态转移方程f[i][j]=(f[i-1][j-1]*(m-j+1)%mod+f[i-1][j])%mod来计算节点i时的方案数。最终将所有节点的方案数累加并取模得到答案。题目来源于牛客网，适合熟悉树结构和动态规划的程序员阅读。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

牛客练习赛-树（染色dp）

(以后刷题还是多总结，排版就丑一点了）

题目描述

链接：https://ac.nowcoder.com/acm/contest/2/B
来源：牛客网

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

输出
复制

备注:

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

题解

懒得排版了，以后题解就写在注释里

#include<bits/stdc++.h>
/*
状态转移方程f[i][j]=(f[i-1][j-1]*(m-j+1)%mod+f[i-1][j])%mod，f[i][j]表示节点i的
节点中如果有j种颜色的话
和力扣的监控那题一样
*/
using namespace std;
#define int