【NOIP2016提高A组集训第14场11.12】随机游走

题目

YJC最近在学习图的有关知识。今天,他遇到了这么一个概念:随机游走。随机游走指每次从相邻的点中随机选一个走过去,重复这样的过程若干次。YJC很聪明,他很快就学会了怎么跑随机游走。为了检验自己是不是欧洲人,他决定选一棵树,每条边边权为1,选一对点s和t,从s开始随机游走,走到t就停下,看看要走多长时间。但是在走了10000000步之后,仍然没有走到t。YJC坚信自己是欧洲人,他认为是因为他选的s和t不好,即从s走到t的期望距离太长了。于是他提出了这么一个问题:给一棵n个点的树,问所有点对(i,j)(1≤i,j≤n)中,从i走到j的期望距离的最大值是多少。YJC发现他不会做了,于是他来问你这个问题的答案。

分析

套路:
首先知道期望是可以累加的,即i通过j去到k的期望,等于i去到j的期望加j去到k的期望。
所以令d[i]表示i的出度,F[i]表示从i到i的父亲的期望,G[i]表示i的父亲到i的期望,j表示i其中任意一个儿子,k表示i的父亲,l表示k其中任意一个儿子,e表示k的父亲。
很容易推出:

[F[i]=dfrac{1}{d[i]}+dfrac{1}{d[i]}sum(1+F[j]+F[i]) ]

[G[i]=dfrac{1}{d[k]}+dfrac{1}{d[k]}(1+G[k]+G[i])+dfrac{1}{d[k]}sum(1+F[l]+G[i]) ]

简化后得

[F[i]=sum{F[j]}+d[i] ]

[G[i]=G[k]+sum{F[l]}+d[k] ]


枚举(i,j)的lca,
记录在以lca为根的子树中,
从lca出发,和到达lca的最大期望值,
为了防止两个最大值的lca不是枚举了lca,记录枚举儿子中的最大值,
取最大次大值比较。

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
const int maxlongint=2147483647;
const int mo=1000000007;
const int N=100005;
using namespace std;
int f[N],g[N];
int n,m,next[N*3],last[N*3],to[N*3],d[N],tot,deep[N],mx[N*2][8],ans;
int bj(int x,int y)
{
	next[++tot]=last[x];
	last[x]=tot;
	to[tot]=y;
}
int dg(int x,int fa)
{
	f[x]=d[x];
	for(int i=last[x];i;i=next[i])
	{
		int j=to[i];
		if(j!=fa)
		{
			deep[j]=deep[x]+1;
			dg(j,x);
			f[x]+=f[j];
		}
	}
}
int dg1(int x,int fa)
{
	int sum1f=0;
	for(int i=last[x];i;i=next[i])
	{
		int j=to[i];
		if(j!=fa)
		{
			g[j]+=g[x]+d[x];
			sum1f+=f[j];
		}
	}
	for(int i=last[x];i;i=next[i])
	{
		int j=to[i];
		if(j!=fa)
		{
			g[j]+=sum1f-f[j];
			dg1(j,x);
		}
	}
}
int dg2(int x,int fa)
{
    for(int i=last[x];i;i=next[i])
    {    
		int j=to[i];
		if(j!=fa)
		{
			dg2(j,x);
			if(mx[j][0]+f[j]>=mx[x][0])
			{
				mx[x][1]=mx[x][0];
				mx[x][0]=mx[j][0]+f[j];
				mx[x][5]=mx[x][4];
				mx[x][4]=j;
			}
			else
			if(mx[j][0]+f[j]>mx[x][1])
			{
				mx[x][1]=mx[j][0]+f[j];
				mx[x][5]=j;
			}
			
			if(mx[j][2]+g[j]>=mx[x][2])
			{
				mx[x][3]=mx[x][2];
				mx[x][2]=mx[j][2]+g[j];
				mx[x][7]=mx[x][6];
				mx[x][6]=j;
			}
			else
			if(mx[j][2]+g[j]>mx[x][3])
			{
				mx[x][3]=mx[j][2]+g[j];
				mx[x][7]=j;
			}
		}
    }
    if(mx[x][4]!=mx[x][6])
    {
    	if(mx[x][0]+mx[x][2]>ans) ans=mx[x][0]+mx[x][2];
	}
    if(mx[x][5]!=mx[x][6])
	{
		if(mx[x][1]+mx[x][2]>ans) ans=mx[x][1]+mx[x][2];
	}
    if(mx[x][4]!=mx[x][7])
    {
    	if(mx[x][0]+mx[x][3]>ans) ans=mx[x][0]+mx[x][3];
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		d[x]++;
		d[y]++;
		bj(x,y);
		bj(y,x);
	}
	deep[1]=1;
	dg(1,0);
	dg1(1,0);
	dg2(1,0);
	printf("%d.00000",ans);
}

原文地址:https://www.cnblogs.com/chen1352/p/9066595.html