https://nanti.jisuanke.com/t/41392
卧槽这题这么水竟然只过了200+,I题那么难,过了400+卧槽,想I去了,结果队友一直WA这题
f[u]为以u为根节点的能输出最大值的概率,一次随机并dfs能得到最大值的概率为tmp= sum{f[v]} /sons[u],如果v无法找到最大值,f[v]=0,如果v就是最远点,那么f[v]=1。
那么k次随机能找到的概率就是 f[u]=1-(1-tmp)^k ,就是所有情况 减去 k次全都找不到的概率。
#include<bits/stdc++.h>
#define maxl 1000010
using namespace std;
const int mod=1e9+7;
int n,cnt;
int ehead[maxl],sons[maxl],dis[maxl],mx[maxl];
long long f[maxl];
struct ed
{
int to,nxt;
}e[maxl*2];
inline void add(int u,int v)
{
e[++cnt].to=v;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}
inline void dfs(int u,int fa)
{
int v;
mx[u]=dis[u];
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(fa==v) continue;
sons[u]++;dis[v]=dis[u]+1;
dfs(v,u);
mx[u]=max(mx[u],mx[v]);
}
}
inline void prework()
{
for(int i=1;i<=n;i++)
ehead[i]=0,sons[i]=0,dis[i]=0,f[i]=0;
int u,v;cnt=0;
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v);add(v,u);
}
dis[1]=1;
dfs(1,1);
}
inline long long qp(long long a,int b)
{
long long ans=1,cnt=a;
while(b)
{
if(b&1)
ans=ans*cnt%mod;
cnt=cnt*cnt%mod;
b>>=1;
}
return ans;
}
inline void solv(int u,int fa)
{
int v;long long sum=0,tmp;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(v==fa) continue;
if(mx[v]==mx[1])
{
solv(v,u);
sum=(sum+f[v])%mod;
}
}
if(sons[u]==0)
f[u]=1;
else
{
tmp=sum*qp(1ll*sons[u],mod-2)%mod;
tmp=((1-tmp)%mod+mod)%mod;
tmp=qp(tmp,sons[u]);
tmp=((1-tmp)%mod+mod)%mod;
f[u]=tmp;
}
}
inline void mainwork()
{
solv(1,1);
}
inline void print()
{
printf("%lld\n",f[1]);
}
int main()
{
while(~scanf("%d",&n))
{
prework();
mainwork();
print();
}
return 0;
}