Codeforces Round #263 (Div. 1) B. Appleman and Tree(给一棵树,每个点为白色或黑色,切断一些边,使得每个连通块有且仅有一个黑点,问划分方案数。)

题意:给一棵树,每个点为白色或黑色,切断一些边,使得每个连通块有且仅有一个黑点,问划分方案数。(n<=1e5)


思路:
dp[u][0]表示没有被包含在黑色连通块中
dp[u][1]表示被包含在黑色连通块中
dp[u][0]=dp[u][0]*dp[v][1]+dp[u][0]*dp[v][0]
dp[u][1]=dp[u][0]*dp[v][1]+dp[u][1]*dp[v][0]+dp[u][1]*dp[v][1]


#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
const int N=1e5+100;
const int MOD=1e9+7;
struct Edge{
    int to,next;
}e[N*2];
int a[N],ans[N],tot,head[N],size[N];
long long dp[N][2];

void init(){
    tot=0;
    memset(head,-1,sizeof(head));
}

void addedge(int from,int to){
    e[tot]=(Edge){to,head[from]};
    head[from]=tot++;
}

void dfs(int u){
    if(size[u]) dp[u][1]=1;
    dp[u][0]=dp[u][1]^1;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        dfs(v);
        long long t1=dp[u][0]*dp[v][1]+dp[u][0]*dp[v][0];
        long long t2=dp[u][0]*dp[v][1]+dp[u][1]*dp[v][0]+dp[u][1]*dp[v][1];
        dp[u][0]=t1%MOD,dp[u][1]=t2%MOD;
    }
}

int main(){
    int n,fa;
    scanf("%d",&n);
    init();
    for(int i=1;i<n;i++){
        scanf("%d",&fa);
        addedge(fa,i);
    }
    for(int i=0;i<n;i++)   scanf("%d",&size[i]);
    dfs(0);
    printf("%lld\n",dp[0][1]);
    return 0;
}
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值