【树形DP】AT_dp_p Independent Set 题解

step 1 题意理解

  • 有一棵有 N N N 个顶点的树,编号为 1 , 2 , ... , N 1,2,...,N 1,2,...,N。
  • Taro 决定将每个顶点涂成白色或黑色。 在这里,不允许将相邻的两个顶点都涂成黑色
  • 找出可以涂色的方式数量,对 1 0 9 + 7 10^9 + 7 109+7 取模。

step 2 样例解释

【样例输入】

3

1 2

2 3

【样例输出】

5

【样例解释】

step 3 做法解释

  1. 考虑与没有上司的舞会 相同的分类方法,对 d p dp dp 数组额外开一维来记录上一层的是否是黑色
  2. 对于每一次转移,都有以下转移方程:
  • { f i , j = f i , j × ( f v , 0 + f v , 1 ) j = 0 f i , j = f i , j × f v , 0 j = 1 \begin{cases} f_{i,j} = f_{i,j}\times (f_{v,0} + f_{v,1} )& j = 0 \\ f_{i,j} = f_{i,j}\times f_{v,0} & j = 1 \end{cases} {fi,j=fi,j×(fv,0+fv,1)fi,j=fi,j×fv,0j=0j=1
  • v v v 是 i i i 的儿子

step 4 AC code

cpp 复制代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod = 1e9 + 7;

ll n,vis[100005],dp[100005][2];
vector<ll> tree[100005];

void dfs(int xx){
	dp[xx][0] = dp[xx][1] = 1;
	//cout << tree[xx].size();
	//cout << xx << endl;
	for(int zz : tree[xx]){
		if(!vis[zz]){
			vis[zz] = 1;
			dfs(zz);
			dp[xx][0] *= dp[zz][0] + dp[zz][1];
			dp[xx][0] %= mod;
			dp[xx][1] *= dp[zz][0];
			dp[xx][1] %= mod;
		}
	}
}

int main(){
	cin >> n;
	for(int i = 1; i < n ;i++){
		int x,y;
		cin >> x >> y;
		tree[y] . push_back(x);
		tree[x] . push_back(y);
	}
	vis[1] = 1;
	dfs(1);
	cout << (dp[1][0] + dp[1][1]) % mod;
	return 0;
}
相关推荐
半桔19 小时前
【STL源码剖析】二叉世界的平衡:从BST 到 AVL-tree 和 RB-tree 的插入逻辑
java·数据结构·c++·算法·set·map
R_.L20 小时前
【项目】 :C++ - 仿mudou库one thread one loop式并发服务器实现(代码实现)
服务器·开发语言·c++
R_.L20 小时前
【项目】 :C++ - 仿mudou库one thread one loop式并发服务器实现(模块划分)
服务器·c++
尚久龙20 小时前
安卓学习 之 SeekBar(音视频播放进度条)
android·java·学习·手机·android studio
塔中妖20 小时前
【华为OD】分割数组的最大差值
数据结构·算法·华为od
人生游戏牛马NPC1号20 小时前
学习 Android (二十二) 学习 OpenCV (七)
android·opencv·学习
weixin_3077791321 小时前
最小曲面问题的欧拉-拉格朗日方程 / 曲面极值问题的变分法推导
算法
RTC老炮21 小时前
webrtc弱网-AlrDetector类源码分析与算法原理
服务器·网络·算法·php·webrtc
孤廖21 小时前
【算法磨剑:用 C++ 思考的艺术・Dijkstra 实战】弱化版 vs 标准版模板,洛谷 P3371/P4779 双题精讲
java·开发语言·c++·程序人生·算法·贪心算法·启发式算法
躯坏神辉21 小时前
c++怎么读取文件里的内容和往文件里写入数据
c++