标签:int Tree dfs Appleman fh mul dp mod
## [CF461B] Appleman and Tree - 树形dp
### Description
给你一棵树,有的点是黑色的有的点是白色的,把树分成若干个联通块使得每个联通块有且仅有一个黑点,问有多少种分法。
### Solution
树形 dp,转移时整体枚举一下边割不割即可
$f[i]$ 表示处理掉 i 子树,并且无法上传,$h[i]$ 表示处理掉 i 子树,并且可以上传
``` cpp
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 5;
const int mod = 1e9 + 7;
int qpow(int p, int q)
{
return (q & 1 ? p : 1) * (q ? qpow(p * p % mod, q / 2) : 1) % mod;
}
int inv(int p)
{
return qpow(p, mod - 2);
}
int n, a[N], f[N], h[N];
vector<int> g[N];
void dfs(int p)
{
int mul_fh = 1;
for (int q : g[p])
{
dfs(q);
(mul_fh *= f[q] + h[q]) %= mod;
}
if (a[p] == 0)
f[p] += mul_fh;
if (a[p])
h[p] += mul_fh;
else
for (int q : g[p])
h[p] += h[q] * mul_fh % mod * inv(f[q] + h[q]) % mod;
f[p] %= mod;
h[p] %= mod;
}
signed main()
{
ios::sync_with_stdio(false);
cin >> n;
for (int i = 2; i <= n; i++)
{
int x;
cin >> x;
++x;
g[x].push_back(i);
}
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
dfs(1);
for (int i = 1; i <= n; i++)
{
// cout << "i=" << i << "\t" << f[i] << "\t" << h[i] << endl;
}
cout << h[1] % mod << endl;
}
标签:int,Tree,dfs,Appleman,fh,mul,dp,mod 来源: https://www.cnblogs.com/mollnn/p/14396579.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。