CodeForces 461B :Appleman and Tree 树形DP

传送门

题目描述

给你一棵树
点之间存在父子关系，除根（1号点）之外有的点是黑色的有的点是白色的
把树分成若干个联通块使得每个联通块有且仅有一个黑点，问有多少种分法
第一行n，表示点的数量 第二行n-1个数字，第i个数字表示i+1号节点的父亲 第三行n个数字，表示点的颜色
一个数字，答案对1e9+7取模

分析

好题啊
$f[u][0/1]$表示以$u$为根节点的联通块是否有黑色块的情况数，然后情况就比较好推了

• $f[u][1]$
  假如$u$节点联通块包含了黑色，那么下面链接有黑色的联通块的时候需要去断开连接，链接没有黑色联通块的时候直接链接，如果不包含，那么只能去链接包含黑色的联通块
  所以状态转移方程为$f[u][1]=(f[u][1]\ast f[v][0]\%mod+f[u][1]\ast f[v][1]\%mod+f[u][0]\ast f[v][1]\%mod)\%mod;$
• $f[u][0]$
  首先$u$节点联通块必然不包含黑色，如果下面链接有黑色的联通块的时候需要去断开连接，链接没有黑色联通块的时候直接链接即可
  所以状态转移方程为$f[u][0]=(f[u][0]\ast f[v][1]\%mod+f[u][0]\ast f[v][0]\%mod)\%mod;$

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod = 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a) {
	char c = getchar(); T x = 0, f = 1; while (!isdigit(c)) {if (c == '-')f = -1; c = getchar();}
	while (isdigit(c)) {x = (x << 1) + (x << 3) + c - '0'; c = getchar();} a = f * x;
}
int gcd(int a, int b) {return (b > 0) ? gcd(b, a % b) : a;}
int h[N],ne[N],e[N],idx;
ll f[N][2];
int n;
int col[N];

void add(int x,int y){
	ne[idx] = h[x],e[idx] = y,h[x] = idx++; 
}

void dfs(int u){
	for(int i = h[u];~i;i = ne[i]){
		int v = e[i];
		dfs(v);
		f[u][1] = (f[u][1] * f[v][0] % mod + f[u][1] * f[v][1] % mod + f[u][0] * f[v][1] % mod) % mod;
		f[u][0] = (f[u][0] * f[v][1] % mod + f[u][0] * f[v][0] % mod) % mod;
	}
}

int main() {
	memset(h,-1,sizeof h);
	read(n);
	for(int i = 2;i <= n;i++){
		int x;
		read(x);
		x++;
		add(x,i);
	}
	for(int i = 1;i <= n;i++) read(col[i]),f[i][col[i]] = 1;
	dfs(1);
	dl(f[1][1]);
	return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/