CF804D Expected diameter of a tree 题解

题目传送门

题目大意： 给出一片森林，每次询问给出两个点，问将这两个点所在的树随机连起来（即随机选两个点连一条边），得到的新树的直径期望长度是多少。

题解

先预处理出每个点到自己所在树的最远点距离，记为 $far[i]$，这个很好求，因为每个点的最远点一定是直径的某一端，所以从两端各跑一次 $dfs$ 即可。

询问时，记两棵树的直径的最大值为 $k$，连接两棵树后，直径要么是 $k$，要么就是连接的两个点 $x,y$ 的 $far[x]+far[y]+1$。

于是先枚举小的那棵树的每一个点 $i$，然后二分在较大的树里找一个点 $j$，恰好满足 $far[i]+far[j]+1>k$，找到这个 $j$ 之后，根据单调性，在较大的树中，$far$ 的值比 $far[j]$ 小的点，他们和点 $i$ 连起来的话得到的 $far[x]+far[y]+1$ 肯定小于等于 $k$，所以他们的贡献就是 $k$，而比 $far[j]$ 大的贡献都是 $far[x]+far[y]+1$，可以维护一个后缀和，然后统计起来即可。

以及询问做一下记忆化就可以过了，最坏情况就是有 $\sqrt{n}$ 棵大小为 $\sqrt{n}$ 的树，那么一共有约 $n$ 种本质不同的询问，每次询问复杂度为 $\sqrt{n}\times {\log }_2(\sqrt{n})$。

代码如下：

#include <cstdio>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;
#define maxn 100010
#define ll long long

int n,m,k;
struct edge{int y,next;};
edge e[maxn<<1];
int first[maxn],len=0;
void buildroad(int x,int y)
{
	e[++len]=(edge){y,first[x]};
	first[x]=len;
}
int dis[maxn],far[maxn],zj[maxn],belong[maxn],id=0,p;
//dis记录每个点到起点的距离，far如上所述，zj即直径，belong记录每个点属于哪棵树
bool vis[maxn];
vector<int> vec[maxn];//记录每棵树的节点
ll *suf[maxn];//每棵树的节点按far排序后，求出他们far的后缀和
void dfs(int x,int fa,int type)
{
	if(type==1)vis[x]=true,belong[x]=id,vec[id].push_back(x);
	else if(dis[x]>far[x])far[x]=dis[x];
	if(dis[x]>dis[p])p=x;
	for(int i=first[x];i;i=e[i].next)
	if(e[i].y!=fa)dis[e[i].y]=dis[x]+1,dfs(e[i].y,x,type);
}
void init(int x,int fa)
{
	dis[x]=0;
	for(int i=first[x];i;i=e[i].next)
	if(e[i].y!=fa)init(e[i].y,x);
}
bool cmp(const int x,const int y){return far[x]<far[y];}
void go(int x)
{
	id++;p=x;
	dfs(x,0,1);init(x,0);//两次dfs求直径
	dfs(p,0,2);init(p,0);//第2,3次dfs时求出far
	dfs(p,0,2);init(p,0);
	zj[id]=far[p];
	sort(vec[id].begin(),vec[id].end(),cmp);
	suf[id]=new ll[vec[id].size()+10];//动态开数组，节省空间
	suf[id][vec[id].size()]=0;//求后缀和
	for(int i=vec[id].size()-1;i>=0;i--)
	suf[id][i]=suf[id][i+1]+far[vec[id][i]];
}
struct par{
	int x,y;
	par(int xx=0,int yy=0):x(xx),y(yy){}
	bool operator <(const par &b)const{return x==b.x?y<b.y:x<b.x;}
};
map<par,double>v;
double ans;
int solve(int x,int y,int ZJ)
{
	int l=0,r=vec[y].size()-1,re=r+1;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(x+far[vec[y][mid]]+1>ZJ)re=mid,r=mid-1;
		else l=mid+1;
	}
	return re;
}

int main()
{
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1,x,y;i<=m;i++)
	scanf("%d %d",&x,&y),buildroad(x,y),buildroad(y,x);
	for(int i=1;i<=n;i++)if(!vis[i])go(i);
	for(int i=1,x,y;i<=k;i++)
	{
		scanf("%d %d",&x,&y);
		x=belong[x];y=belong[y];
		if(x==y){printf("-1\n");continue;}
		if(vec[x].size()>vec[y].size())swap(x,y);
		if(v.find(par(x,y))!=v.end()){printf("%.10lf\n",v[par(x,y)]);continue;}
		ans=0.0;
		for(int i=0;i<vec[x].size();i++)
		{
			p=solve(far[vec[x][i]],y,max(zj[x],zj[y]));
			//下面这行一定不能漏了1.0*
			ans+=1.0*p*max(zj[x],zj[y])+1.0*(vec[y].size()-p)*(far[vec[x][i]]+1)+1.0*suf[y][p];
		}
		ans/=1.0*vec[x].size();ans/=1.0*vec[y].size();
		printf("%.10lf\n",ans);v[par(x,y)]=ans;
	}
}