“今日头条杯”首届湖北省大学程序设计竞赛 D. Who killed Cock Robin（选择联通子图的方案数，树形dp）

题意：

给定n个点的树，求选出一个联通子图的方案数，答案对1e9+7取模。

数据范围：n<=2e5

解法：

$$\begin{gathered} 设d[x]以x为根的子树中,选择点x之后,连通子图的方案数, \\ 设v是x的子节点,那么转移方程为: \\ 1.如果选择点v,那么方案数为d[x]\ast d[v]. \\ 2.如果不选择点v,那么d[x]. \\ 综上得转移方程为:d[x]=d[x]\ast (d[v]+1). \end{gathered}$$

code：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
const int mod=1e7+7;
vector<int>g[maxm];
int d[maxm];
int n;
int ans;
void dfs(int x,int fa){
    d[x]=1;
    for(int v:g[x]){
        if(v==fa)continue;
        dfs(v,x);
        d[x]=d[x]*(d[v]+1)%mod;
    }
    ans=(ans+d[x])%mod;
}
signed main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    dfs(1,1);
    cout<<ans<<endl;
    return 0;
}

---