题目:http://codeforces.com/problemset/problem/543/D
题意:给你一棵树,一开始边都是0,可以使任意的边变成1,对于每一个根节点求使得它到其他任一点的路径上只有一条0边的方案数。
假设只求一个根,f[u]=∏(s[v]+1)
然后移动根节点这样就可以通过遍历一遍树得到所有点的答案了。
s[v]=(s[u]/(s[v]+1)+1)*s[v] 这样当前根和根的其他子树就变成v的子树了(前面那坨就是它的贡献。。
(看起来是这样没错。。但是不能求逆元。因为s[u]可能为0 。。所以维护前缀和后缀bfs一遍就好了。。
#include<cstring>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#define rep(i,l,r) for (int i=l;i<=r;i++)
#define down(i,l,r) for (int i=l;i>=r;i--)
#define clr(x,y) memset(x,y,sizeof(x))
#define maxn 200500
#define maxk 69
#define inf 0x7fffffff
#define ll long long
#define mm 1000000007
using namespace std;
struct data{int obj,pre;
}e[maxn*];
ll g[maxn],s[maxn],ans[maxn],pr[maxn],sc[maxn];
int head[maxn],a[maxn],vis[maxn],n,tot;
int read(){
ll x=,f=; char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-; ch=getchar();}
while (isdigit(ch)){x=x*+ch-''; ch=getchar();}
return x*f;
}
void insert(int x,int y){
e[++tot].obj=y; e[tot].pre=head[x]; head[x]=tot;
}
ll ni(ll x){
ll y=mm-,ans=;
while (y){
if (y&) ans=ans*x%mm;
x=x*x%mm;
y>>=;
}
return ans;
}
ll dfs(int u,int f){
s[u]=;
for (int j=head[u];j;j=e[j].pre){
int v=e[j].obj;
if (v==f) continue;
s[u]=s[u]*(dfs(v,u)+1LL)%mm;
}
return s[u];
}
void go(){
g[]=;
queue<int> q; clr(vis,); q.push();
while (!q.empty()){
int u=q.front(); q.pop(); vis[u]=;
int sum=;
for (int j=head[u];j;j=e[j].pre){
int v=e[j].obj;
if (!vis[v]) a[++sum]=v;
}
sort(a+,a++sum);
pr[]=; rep(i,,sum) pr[i]=pr[i-]*(s[a[i]]+)%mm;
sc[sum+]=g[u]; down(i,sum,) sc[i]=sc[i+]*(s[a[i]]+)%mm;
rep(i,,sum) {
s[a[i]]=(pr[i-]*sc[i+]%mm+)%mm*s[a[i]]%mm;
g[a[i]]=(pr[i-]*sc[i+]%mm+)%mm;
q.push(a[i]);
}
}
}
int main(){
n=read();
rep(i,,n) {
int x=read(); insert(x,i); insert(i,x);
}
dfs(,);
go();
rep(i,,n-) printf("%lld ",s[i]%mm); printf("%lld\n",s[n]%mm);
return ;
}