2023河南萌新联赛第（六）场：河南理工大学 C 旅游

$题目大意:$
$给定一棵n个节点的树，点之间有长度为1的路径，求起始点为任意点时走的路径长度的期望最大值$$（不会走重复的边，且走每一条边的概率一致)$
$题目分析:$
$可以考虑换根DP,设d{p}_x表示从x开始向下走经过节点数的期望$
$易知d{p}_x=1+\frac{1}{|so{n}_x|}{\sum }_{s\in so{n}_x}d{p}_s$
$接下来，可以设g_x表示以x为起始点的期望$
$由于我们计算dp数组时必以一点为根节点，不妨为1,记算完dp数组后，那么g_1=d{p}_1$
$接下来便考虑换根DP,对于v，我们假设已知v的父亲的值g_u，首先考虑v的子树内的贡献，可知贡献为{\sum }_{s\in so{n}_v}d{p}_s,由dp式子的变形可知{\sum }_{s\in so{n}_v}d{p}_s=(d{p}_v-1)\ast \left|so{n}_v\right|,显然可以O(1)求出$
$再考虑v的父节点的贡献，可知贡献为\frac{|so{n}_u+1|\ast g_u-d{p}_v-1}{|so{n}_u|}$
$最后还有v对自己的贡献1$
$故可以得出转移式g_v=\frac{1}{|so{n}_v+1|}\ast ((d{p}_v-1)\ast \left|so{n}_v\right|+\frac{|so{n}_u+1|\ast g_u-d{p}_v-1}{|so{n}_u|})+1$
$最后答案便为ma{x}_{i=1}^n{g}_i-1$
$时间复杂度为O(n)$
参考代码:

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int M=100007;

int n;
vector<int> G[M];
double dp[M],g[M];

void DFS1(int now,int fa){
	int v;
	double sum=0;
	for(int i=0;i<G[now].size();i++){
		v=G[now][i];
		if(fa==v) continue;
		DFS1(v,now);
		sum+=dp[v];
	}
	if(now==1)dp[now]=1+sum/G[now].size();
	else if(G[now].size()==1) dp[now]++;
	else dp[now]=1+sum/(G[now].size()-1);
}

void DFS2(int now,int fa){
	int v;
	double sum;
	for(int i=0;i<G[now].size();i++){
		v=G[now][i];
		if(fa==v) continue;
		sum=0;
	    sum=sum+(dp[v]-1)*(G[v].size()-1);
	    if(G[now].size()>1) {
	    	sum=sum+(g[now]*G[now].size()-dp[v]-1)/(G[now].size()-1);
		}else {
			sum=sum+1;
		}
	    g[v]=sum/(G[v].size())+1;
		DFS2(v,now);
	}
}
int main(){
	ios::sync_with_stdio(false);
	int a,b;
	cin>>n;
	for(int i=1;i<n;i++){
		cin>>a>>b;
		G[a].push_back(b);
		G[b].push_back(a);
    }
	DFS1(1,-1);g[1]=dp[1];
//	for(int i=1;i<=4;i++){
//		cout<<dp[i]<<" ";
//	}
//	cout<<"\n";
	DFS2(1,-1);
//	for(int i=1;i<=4;i++){
//		cout<<g[i]<<" ";
//	}
	double anq=0;
	for(int i=1;i<=n;i++) anq=max(anq,g[i]);
	cout<<fixed<<setprecision(3)<<anq-1<<"\n";
}