牛客多校3 G Coloring Tree (DFS+BFS)

这是一道关于树染色问题的算法题,目标是找到颜色度为d的染色方案数。题目要求同一颜色的节点间最短距离至少为d。通过定义F(D)表示颜色度大于等于D的方案数,可以得出颜色度为D的方案数为F(D+1) - F(D)。解决方案采用DFS+BFS策略,遍历树的节点并计算子树中已染色节点的数量,从而更新方案数。

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

题意:
给你一个由n个节点n-1条边组成的树,之后你可以在每个节点上面染颜色,定义一个颜色度,表示的是相同颜色最近的距离是多少,现在想问你颜色度为d的染色方案数是多少。
思路
相同颜色最近的距离是D,换句话说就是一个节点在距离D内都不能有相同的颜色。
那么我们定义F(D)表示的是颜色度大于等于D的时候的方案数,那么的话,颜色度等于D的方案数其实就是 F(D+1) - F(D),知道这个之后我们就可以求F(D)了,具体怎么求F(D),我们广搜所有节点,对于我们当前广搜的节点我们深搜看他的子树下面有多少个已经被染色的节点,如果有一个我们就让他减一,之后方案数相乘就是答案上代码把:
代码

#include <bits/stdc++.h>
#define mod 1000000007
using namespace std;
const int maxn = 5000 + 10;
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值