21天好习惯第一期－2 DFS

最新推荐文章于 2024-06-17 19:30:00 发布

原创最新推荐文章于 2024-06-17 19:30:00 发布 · 2.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#1024程序员节

本文解析了如何利用DFS序列和动态规划解决一个关于在树中给每个节点涂色的问题，强调了两种策略：重复颜色沿路径保持一致和使用剩余未用颜色。通过状态转移方程展示了如何计算使用不同颜色的节点组合。

思路：
感觉和dfs序联系不大，画个图后发现涂颜色只有两种方案(引用每日一题的图片)：
在这里插入图片描述
1.使用用过的颜色，但是一定要和它处理过的父结点或者儿子结点涂一样的颜色，只有这样才能保证一样颜色的点在一个路径上；
2。使用没用过的颜色，已经用了j-1种颜色，所以还剩k-j+1种颜色可以涂
可以用dp来求解：
状态：dp[i][j],处理到第i个点时用了j种颜色。
状态转移方程：dp[i][j]=(dp[i-1][j]+dp[i-1][j-1]*(k-j+1))%mod.

代码：

#include<bits/stdc++.h>
#define ll long long 
#define  js  ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;
const int mod=1e9+7;
ll dp[305][305],ans=0;
int n,k;
int main() {
    js;
    cin>>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-1][j]+dp[i-1][j-1]*(k-j+1))%mod;
    for(int i=1;i<=k;++i)    ans=(ans+dp[n][i])%mod;
    cout<<ans<<endl;
    return 0;
}