Codeforces Round #428 (Div. 2) C. Journey(【dfs】求期望值)

題目鏈接:http://codeforces.com/contest/839/problem/C

【中文題意】有n個城市,n-1條邊(保證圖連通),從1城市出發,走過的城市不可以再走,每兩個城市之間的距離爲1,走過的城市不可以再走,求出走過的距離的期望值。
【思路分析】首先就是期望值,期望值爲每個距離乘以其概率,然後相加。然後我們就可以從城市1開始dfs,dfs到葉子節點的時候加上距離乘以期望值,這裏爲了減少精度損失,我dfs時候記錄的爲概率的倒數。
【AC代碼】

#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<stack>
#include<queue>
#include<set>
using namespace std;
#define LL long long

const int maxn = 100000 + 10;
int n;
vector<int>G[maxn];
bool book[maxn];
double sum = 0;
void dfs(int u,int len,double s)
{
    int nn=G[u].size();
    int cnt=0;
    for(int i=0;i<nn;i++)
    {
        int v=G[u][i];
        if(!book[v])
        {
            cnt++;
        }
    }
    if(cnt==0)
    {
        sum += (double)len/s;
        //printf("**%d %lf\n",len,s);
        return;
    }
    for(int i=0;i<nn;i++)
    {
        int v=G[u][i];
        if(!book[v])
        {
            book[v]=1;
            dfs(v,len+1,s*(double)cnt);
        }
    }
    return ;
}

int main()
{
    while(~scanf("%d",&n))
    {
        int u,v;
        for(int i = 1; i <= n; i++)G[i].clear();
        for(int i = 1; i < n; i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        memset(book,0,sizeof(book));
        book[1]=1;
        sum=0;
        dfs(1,0,1.0);
        printf("%.15lf\n",sum);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章