CF543D Road Improvement

LXXXVIII.CF543D Road Improvement

常规换根DP题。

我们可以设$f_i$表示以$i$为根的子树中的方案数。则有转移式

$$f_i=\prod\limits_{j\in son_i}(f_j+1)$$

其中$+1$的意思是将边$(i,j)$留作坏边。

显然换根DP就很好实现了。

但一个问题就是换根的时候可能会出现除数为$0$的情形；故我们不能直接简单地除以逆元。所以我们须要预处理出来前缀积与后缀积，这样就能在转移过程中避免逆元辣。

时间复杂度$O(n)$。

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
int n,f[200100];
vector<int>v[200100],pre[200100],suf[200100];
void dfs1(int x){
	f[x]=1;
	for(auto y:v[x])dfs1(y),f[x]=1ll*f[x]*(f[y]+1)%mod;
	if(v[x].empty())return;
	pre[x].resize(v[x].size()),suf[x].resize(v[x].size());
	pre[x][0]=1;for(int i=1;i<v[x].size();i++)pre[x][i]=1ll*pre[x][i-1]*(f[v[x][i-1]]+1)%mod;
	suf[x].back()=1;for(int i=(int)v[x].size()-2;i>=0;i--)suf[x][i]=1ll*suf[x][i+1]*(f[v[x][i+1]]+1)%mod; 
}
void dfs2(int x,int qwq=1){
	for(int i=0;i<v[x].size();i++){
		int y=v[x][i];
		int tmp=1ll*pre[x][i]*suf[x][i]%mod*qwq%mod;
		f[y]=1ll*f[y]*(tmp+1)%mod;
		dfs2(y,tmp+1);
	}
}
int main(){
	scanf("%d",&n);
	for(int i=2,x;i<=n;i++)scanf("%d",&x),v[x].push_back(i);
	dfs1(1),dfs2(1);
	for(int i=1;i<=n;i++)printf("%d ",f[i]);
	return 0;
}