小红树上染色 -树形dp

最新推荐文章于 2024-07-12 18:02:43 发布

.y.a.o.

最新推荐文章于 2024-07-12 18:02:43 发布

阅读量250

点赞数 2

文章标签：深度优先算法 c++

本文链接：https://blog.csdn.net/m0_75129175/article/details/135908578

版权

题面

分析

相邻不能有都是白色的，可以通过树形dp，设 $d p [i] [j]$ 为表示编号为 $i$ 的节点是否染红，（染红 $j$ 为1，否则为0)。就可以实现状态方程。

代码

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

const int N = 1e5 + 10;
const int mod = 1e9 + 7;

vector<int> adj[N];
ll dp[N][2];

void dfs(int u, int fa) {
    dp[u][0] = dp[u][1] = 1;
    for(auto j: adj[u]) {
        if(j == fa) continue;
        dfs(j, u);
        dp[u][0] = dp[u][0] * dp[j][1] % mod;
        dp[u][1] = (dp[u][1] * (dp[j][0] + dp[j][1]) % mod) % mod;
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int n;
    cin >> n;
    for(int i = 0; i < n - 1; i ++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    dfs(1, 0);
    cout << (dp[1][0] + dp[1][1])% mod << "\n";
}