蔚来杯“2022牛客暑期多校训练营5 D：Birds in the tree

最新推荐文章于 2024-07-25 11:27:31 发布

sliver-awa

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量82

点赞数 1

文章标签：算法图论动态规划 c++

本文链接：https://blog.csdn.net/Penny_rd/article/details/126124510

版权

蔚来杯“2022牛客暑期多校训练营5 D：Birds in the tree

原题：https://ac.nowcoder.com/acm/contest/33190/D

题面

One day, when NIO was walking under the tree, he found that there are birds standing on the tree, and each bird is standing on one leaf. He wondered about in how many sub-trees form by connected vertex-induced subgraphs from the original tree, all birds are in the same gender. The number would be very large, just mod $10^9+7$ in the output.

大意

给定一个有 $n$ 个节点的树，每个节点有一个颜色，分别有 $0$ 和 $1$ 。求这颗树里有多少连通的子图，其中度数为 $1$ 的节点的颜色相同。注意取模

题解

用树形 $d p$ ，设 $dp_x,_c$ ，表示以 $x$ 为根的子树内 $x$ 被选择，除 $x$ 外叶节点颜色为 $c$ 的方案数。
则其方案数为从 $dp\scriptstyle son,c$ 中从0开始取，从而得到方案数可表示为：
$dp\scriptstyle x,c$ $=$ $\displaystyle\prod_{s \in son_x}$ $(1+dp\scriptstyle s,c)$

设 $x$ 的颜色为 $col_x$ ，最终答案先加上以 $x$ 为叶节点的方案数，加上只有 $x$ 的图，再加上不以 $x$ 为叶节点的方案，去掉其中与子节点重复的方案与只有 $x$ 的图。
所以，得到
$an s =$ $\displaystyle\sum_{x}$ $(\displaystyle\sum_{s \in son_x} dp\scriptstyle s,col_x$ $+ 1)$ + $\displaystyle\sum_{x}\displaystyle\sum_{c \in {0,1}} dp\scriptstyle x,c$ $-$ $col_x==c)$

接着可以整理一下公式，写出代码

参考代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=3*1e5+5;
int dp[maxn],ans=0;
ll mod=1e9+7;
string s;
vector<int> e[maxn];
void dfs(int u,int fa,char c){
    ll mul=1;
    ll sum=0;
    for(auto i:e[u]){
        if(i==fa) continue;
        dfs(i,u,c);
        sum=(sum+dp[i])%mod;
        mul=mul*(dp[i]+1)%mod;
    }
    dp[u]=(mul-(s[u]!=c)+mod)%mod;
    if(s[u]==c) ans=(ans+dp[u])%mod;
    else ans=(ans+dp[u]-sum+mod)%mod;
}
int main()
{
    int n,x,y;
    while(cin>>n)
    {
        ans=0;
        cin>>s;
        for(int i=1;i<=n-1;i++)
        {
            cin>>x>>y;
            x--;
            y--;
            e[x].push_back(y);        
            e[y].push_back(x);
        }
        dfs(1,-1,'0');
        dfs(1,-1,'1');
        cout<<ans;
    }
    return 0;
}