小红树上染色 -树形dp
题面
分析
相邻不能有都是白色的,可以通过树形dp,设 d p [ i ] [ j ] dp[i][j] dp[i][j] 为表示编号为 i i i 的节点是否染红,(染红 j j 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";
}