2020威海-Rencontre-(思维+贡献+数论期望)

C

题意：
就是给你一个树，然后有3个科学家，每个科学家有mi个想去的点。然后对于每次每个科学家会选择一个点，求出来一个点使得这三个科学家集合到这个点的总距离最小。然后对于所有的可能的这些距离全部加起来，最后再除以一共有多少种可能性。

思考：
说实话，当时看完题意就感觉又是那种假期望题，和上次济南一样，肯定有什么规律可循。对于每次是选3个点，求最小距离的话我只会枚举点后lca求出来，但是时间绝对炸了。因为每个科学家最多有n个想去的地方，这光枚举三个点都nnn的复杂度了，所以肯定不行。对于树上边距离很明显可以想到每条边的贡献，但是这贡献也没法求啊，而且一般求一个边的贡献，就是看上面有多少点和下面有多少点，就是两种点而言的，但是这个三种点而言啊。其实你画图可以发现，对于任意3点a,b,c，他们集合的最短距离其实就是(dis(a,b)+dis(a,c)+dis(b,c))/2。只要你能发现这个，就好办了，对于每条边，看看上面a点集多少下面c点集多少，同理上面c和下面a，同理ac,bc也是这样。这样就转化成两种点集之间的事情了把，如果全部加来再最后除以方案数肯定炸ll，所以每次都除一下当前的可能性。因为期望是可以拆分和累加的，不同的期望最后再加起来就行了。
对于为什么是他们集合的最短距离(dis(a,b)+dis(a,c)+dis(b,c))/2，其实你画出来一个图，用手模拟一下就会发现就是这样的。

代码：

int T,n,m,k;
int va[N];
int cnt[N];
int siz[N][5];
db ans = 0;

vector<PII > e[N];

void get(int now,int p)
{
	for(auto t:e[now])
	{
		int spot = t.fi;
		if(spot==p) continue;
		get(spot,now);
		siz[now][1] += siz[spot][1];
		siz[now][2] += siz[spot][2];
		siz[now][3] += siz[spot][3];
	}	
}

void dfs(int now,int p)
{
	for(auto t:e[now])
	{
		int spot = t.fi,w = t.se;
		if(spot==p) continue;
		dfs(spot,now);
		for(int i=1;i<=3;i++)
		{
			for(int j=i+1;j<=3;j++) //一定要注意都用db转话一下
			{
				ans += (db)(siz[1][i]-siz[spot][i])*(db)siz[spot][j]*w/(db)(cnt[i]*cnt[j])/2.0;
				ans += (db)(siz[1][j]-siz[spot][j])*(db)siz[spot][i]*w/(db)(cnt[i]*cnt[j])/2.0; 
			}
		}
	}
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		e[a].pb({b,c});
		e[b].pb({a,c});
	}
	for(int i=1;i<=3;i++)
	{
		cin>>cnt[i];
		for(int j=1;j<=cnt[i];j++)
		{
			int x;
			cin>>x;
			siz[x][i]++;
		}
	}
	get(1,0);dfs(1,0);
	printf("%.8lf",ans);
	return 0;
}

总结：
多多积累经验把，多画画图思考思考，猜猜结论。