luogu5007 DDOSvoid 的疑惑 (树形dp)

我们来算每个点出现在的集合的个数

设f[i]为i出现的集合个数,g[i]是只选子树i 可以有多少种选法

那就有$g[i]=1+\prod\limits_{j是i的孩子}{g[j]} , f[i]=f[fa[i]]*\prod\limits_{j是i的兄弟}{f[j]}$

这个兄弟的积可以直接用一个逆元 或者是做前缀积和后缀积然后乘起来，但千万不要一边记着后缀积一边又进子树dfs 你孩子全都给你改没了（摔

那我们肯定能不带log就不带嘛

 #include<bits/stdc++.h>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 typedef pair<int,int> pa;
 const int maxn=1e6+,P=1e8+;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int eg[maxn*][],egh[maxn],ect;
 int N,T,stk[maxn];
 ll choi[maxn],dp[maxn];
 ll pre[maxn],ans;

 inline void adeg(int a,int b){
     eg[++ect][]=b,eg[ect][]=egh[a],egh[a]=ect;
 }

 void dfs1(int x,int f){
     choi[x]=;
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];if(b==f) continue;
         dfs1(b,x);
         choi[x]=choi[x]*choi[b]%P;
     }
     choi[x]++;choi[x]%=P;
 }

 void dfs2(int x,int f){
     ans+=dp[x]*(T?x:)%P,ans%=P;
     int n=;
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];if(b==f) continue;
         stk[++n]=b;
     }
     pre[n+]=;
     for(;n;n--) pre[n]=pre[n+]*choi[stk[n]]%P;
     ll a=dp[x];
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];if(b==f) continue;
         ++n;dp[b]=a*pre[n+]%P;
         a*=choi[b],a%=P;
     }
     for(int i=egh[x];i;i=eg[i][]){
         int b=eg[i][];
         if(b==f) continue;
         dfs2(b,x);
     }
 }

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     N=rd(),T=rd();
     for(i=;i<N;i++){
         int a=rd(),b=rd();
         adeg(a,b);adeg(b,a);
     }
     dfs1(,);dp[]=;dfs2(,);
     printf("%lld\n",(ans+P)%P);
     return ;
 }