JZOJ4890. 随机游走

题目大意

给定一棵$n$个节点的树，在一个节点时，下一步会等概率的选择一个相邻的点走过去。现要选择一个起点$s$和一个终点$t$，求最大的期望距离。

Data Constraint
$n\leq100000$

题解

假如我们知道了一条边$(u,v)$，从$u\rightarrow v$的期望距离和$v\rightarrow 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 ;
}

以上.