经典树形DP—小红树上染色问题

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目如下：

很显然这是一个树形DP问题  

直接先后序遍历整棵树。

void dfs(int x,int fa){ x为当前节点编号，fa为当前节点的父亲节点编号

    dp[x][0]=dp[x][1]=1;//假如只有一个节点 那方案就只有一个，就是染他本身。

    for(auto i:g[x]){
        if(i==fa)continue;如果遍历到当前节点是自己的父亲，则结束本轮，继续往下搜。
        dfs(i,x);

dp...

}

状态转移的关键方程为：

  1. dp[x][0]=dp[i][1]*dp[x][0]%mod;

当以x为父亲节点的树不染红色的方案数是--子树i染红色的方案*自身(一直更新自身所延伸的子树的方案数) 乘法原理

        2. dp[x][1]=dp[x][1]*(dp[i][0]+dp[i][1])%mod;

当以x为父亲节点的树染红色的方案数是--子树i染红色的方案*自身+子树i不染红色的方案*自身。

代码如下：

#include<bits/stdc++.h>
using namespace std;
vector<int>g[1000000];
int vis[1000000];
long long dp[101010][2]; //dp[i][0]代表不染红色的节点i号的子树的方案数，dp[i][1]代表染红色节                                                
                           点i号的子树的方案数
int mod = 1e9+7;
void dfs(int x,int fa){

    dp[x][0]=dp[x][1]=1;//假如只有一个节点 那方案就只有一个，就是染他本身。

    for(auto i:g[x]){
        if(i==fa)continue;
        dfs(i,x);
        dp[x][0]=dp[i][1]*dp[x][0]%mod;
        dp[x][1]=dp[x][1]*(dp[i][0]+dp[i][1])%mod;
    }
}
int main(){
    int n;
   
    cin>>n;
    for(int i=1;i<n;i++){
        int x,y;
        cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
        
    }
    dfs(1,0);
    cout<<(dp[1][0]+dp[1][1])%mod;
}

谢谢观看