Tree（树形DP）

Tree（树形DP）

不太熟，记录一下思路

传送门

思路：对于树上某一个点的连通点集的数量，我们可以以它为根进行树上$dp$。

根据乘法原理：$dp[u]=dp[u]\times\prod\limits_{v\in son}(dp[v]+1)$

举个例子：

以$a$为根，显然$dp[b]=dp[c]=1,$通过边$(a,b)$产生的贡献是$dp[b]+1$。这个加1表示不选$b$的子集，然后每个$dp[b]+1$又会对应$dp[c]+1$种情况。

因此可以用乘法原理求得。

但是此题要求我们求出所有结点的连通点集数量，显然一个个$dfs$会超时。

所以我们考虑由已知的$dp[fa]$，进行状态转移。

显然若$u$的$fa$的$dp[fa]$已知，当我们以$u$为根结点求答案时，我们只需用$dp[u]\times (u的$儿子$fa$的贡献)，因为$dp[fa]$存在重复计算$dp[u]$的情况，所以我们应该先把$dp[fa]$除以$dp[u]+1$还原，然后$\dfrac{dp[fa]}{dp[u]+1}+1$即是$fa$的贡献。

综上：状态转移方程为：$dp[u]=dp[u]\times(\dfrac{dp[fa]}{dp[u]+1}+1)$

另外由于此题数据很大，需要取模，而且我们进行状态转移的时候存在除法的情况。

显然我们需要对$dp[u]+1=0$的情况进行特判。

我们需要再开一个数组来维护特判$0$之后的$\dfrac{dp[fa]}{dp[u]+1}$，我们记为$dp1[fa]$。

然后我们再以$fa$为之前的根结点进行状态转移，注意不能算$u和fa[fa]$。

因为$fa[fa]$已经转移过了，即($dp1[fa]+1)$包含了$dp[fa[fa]]$的情况，就相当于隔断了，以$fa$原来的根转移。

时间复杂度：$O(2nlogn)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5,M=1e6+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a) memset(a,0,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first 
#define se second
ll dp[N],dp1[N],ans[N];
int n,Fa[N];
vector<int>e[N];
inline void read(int &x){ 
	x=0;int w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
	for(;ch>='0'&&ch<='9';ch=getchar())
		x=(x<<3)+(x<<1)+(ch&15);
	x*=w; 
}
ll ksm(ll a,ll n){
	ll ans=1;
	while(n){
		if(n&1) ans=ans*a%mod;
		a=a*a%mod;
		n>>=1;
	}
	return ans;
}
void dfs(int u,int fa){
	dp[u]=1,Fa[u]=fa;
	for(auto v:e[u]){
		if(v==fa) continue;
		dfs(v,u);
		dp[u]=dp[u]*(dp[v]+1)%mod;
	}
}
void dfs1(int u,int fa){
	if(!fa) ans[u]=dp[u];
	else {
		ll tmp;
		if((dp[u]+1)%mod==0){
			 tmp=dp1[fa]+1; 
			for(auto v:e[fa]){
				if(v==u||v==Fa[fa]) continue; 
				tmp=tmp*(dp[v]+1)%mod;
			}
			//printf("%lld\n",tmp);
			dp1[u]=tmp;
			ans[u]=dp[u]*(tmp+1)%mod; 
		}
		else tmp=ans[fa]*ksm(dp[u]+1,mod-2)%mod,ans[u]=dp[u]*(tmp+1)%mod;
	}
	for(auto v:e[u]){
		if(v==fa) continue;
		dfs1(v,u);
	}
} 
int main(){
	int  n;
	read(n);
	for(reg int i=1;i<n;i++){
		int u,v;
		read(u),read(v);
		e[u].push_back(v),e[v].push_back(u);
	}
	dfs(1,0);
	dfs1(1,0);
	for(int i=1;i<=n;i++) printf("%lld\n",ans[i]);
	return 0;
}