经典树状dp 牛客周赛Round 30 E题 小红树上染色题解

第一步、构造树

#include<bits/stdc++.h>
using namespace std;

#define int long long 
#define pb push_back

vector<int> g[200020];
signed main() {
	int n;
	cin >> n;
	for (int i=1; i<=n; i++) {
		int x,y;
		cin >> x >> y;
// g[x]存储 x结点的邻点
		g[x].pb(y);
		g[y].pb(x);
	}
}

第二步、遍历树

void dfs(int x, int pr) {
	for (auto i : g[x]) {
		// 遍历到父亲结点直接跳过
		if (i == x) continue;
		dfs(i, x); // i是当前结点 x是父亲结点 
	}
}

第三步、dp推导

dp[i][0] 代表 i 结点 不染红的方案数

dp[i][1] 代表 i 结点 染红的方案数

从孩子的状态推导父亲的状态

        若 i 号结点没有染红， 则它的孩子必须都染红

        若 i 号结点染红， 则它的孩子可以染红也可以不染红

vector<int> g[200020];
int dp[101010][2];
void dfs(int x, int pr) {
	dp[x][0] = dp[x][1] = 1;
	for (auto i : g[x]) {
		// 遍历到父亲结点直接跳过
		if (i == pr) continue;
		dfs(i, x); // i是当前结点 x是父亲结点 
		dp[x][0] = dp[x][0] * dp[i][1] % mod;
		dp[x][1] = dp[x][1] * (dp[i][0] + dp[i][1]) % mod;
	}
}

第四步、输出根结点（1）两种情况之和即可，AC代码如下

#include<bits/stdc++.h>
using namespace std;

#define int long long 
#define pb push_back
#define mod 1000000007

vector<int> g[200020];
int dp[101010][2];
void dfs(int x, int pr) {
	dp[x][0] = dp[x][1] = 1;
	for (auto i : g[x]) {
		// 遍历到父亲结点直接跳过
		if (i == pr) continue;
		dfs(i, x); // i是当前结点 x是父亲结点 
		dp[x][0] = dp[x][0] * dp[i][1] % mod;
		dp[x][1] = dp[x][1] * (dp[i][0] + dp[i][1]) % mod;
	}
}

signed main() {
	int n;
	cin >> n;
	for (int i=1; i<n; i++) {
		int x,y;
		cin >> x >> y;
		g[x].pb(y);
		g[y].pb(x);
	}
	dfs(1,0);
	cout << (dp[1][0]+dp[1][1])%mod;
}