题目大意
给定一棵
n
个节点的树,在一个节点时,下一步会等概率的选择一个相邻的点走过去。现要选择一个起点
Data Constraint
题解
假如我们知道了一条边
(u,v)
,从
u→v
的期望距离和
v→u
的期望距离,我们就能比较简单的计算出树上过某一点的最长期望距离是多少。
所以只要两边DFS预处理那两个期望距离即可。求最长路的过程类似找树的直径。
SRC
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std ;
#define N 100000 + 10
#define M 1000 + 10
const double eps = 1e-8 ;
int Node[2*N] , Next[2*N] , Head[N] , tot ;
double eu[N] , ed[N] , f[N] , g[N] ;
int Son[N] ;
int n ;
double ans ;
void link( int u , int v ) {
Node[++tot] = v ;
Next[tot] = Head[u] ;
Head[u] = tot ;
}
void DFS1( int x , int F ) {
double sum = 0 ;
for (int p = Head[x] ; p ; p = Next[p] ) {
if ( Node[p] == F ) continue ;
DFS1( Node[p] , x ) ;
sum += eu[Node[p]] ;
}
eu[x] = sum + Son[x] ;
}
void DFS2( int x , int F ) {
double sum = 0 ;
for (int p = Head[x] ; p ; p = Next[p] ) {
if ( Node[p] == F ) continue ;
sum += eu[Node[p]] ;
}
for (int p = Head[x] ; p ; p = Next[p] ) {
if ( Node[p] == F ) continue ;
ed[Node[p]] = sum - eu[Node[p]] + Son[x] + ed[x] ;
DFS2( Node[p] , x ) ;
}
}
void DFS( int x , int F ) {
int ff = 0 , fg = 0 ;
double sf = 0 , sg = 0 ;
for (int p = Head[x] ; p ; p = Next[p] ) {
if ( Node[p] == F ) continue ;
DFS( Node[p] , x ) ;
if ( f[Node[p]] + eu[Node[p]] > f[x] ) {
sf = f[x] ;
f[x] = f[Node[p]] + eu[Node[p]] ;
ff = Node[p] ;
} else if ( f[Node[p]] + eu[Node[p]] > sf ) sf = f[Node[p]] + eu[Node[p]] ;
if ( g[Node[p]] + ed[Node[p]] > g[x] ) {
sg = g[x] ;
g[x] = g[Node[p]] + ed[Node[p]] ;
fg = Node[p] ;
} else if ( g[Node[p]] + ed[Node[p]] > sg ) sg = g[Node[p]] + ed[Node[p]] ;
}
if ( fg == ff ) {
ans = max( ans , max( f[x] + sg , sf + g[x] ) ) ;
} else ans = max( ans , f[x] + g[x] ) ;
}
int main() {
freopen( "rw.in" , "r" , stdin ) ;
freopen( "rw.out" , "w" , stdout ) ;
scanf( "%d" , &n ) ;
for (int i = 1 ; i < n ; i ++ ) {
int u , v ;
scanf( "%d%d" , &u , &v ) ;
link( u , v ) ;
link( v , u ) ;
Son[u] ++ , Son[v] ++ ;
}
DFS1( 1 , 0 ) ;
DFS2( 1 , 0 ) ;
DFS( 1 , 0 ) ;
printf( "%.5lf\n" , ans ) ;
return 0 ;
}
以上.