2019徐州网络赛 J Random Access Iterator

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;
}