Codeforces 839C Journey

题目链接：

http://codeforces.com/contest/839/problem/C

题解：

昨晚脑子一抽，算了半天的平均期望。。。

代码：

#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
const int maxn = 1e5+10;
vector<int> p[maxn];
vector<int> l;
int visited[maxn];
int dis[maxn];
double EX[maxn];
double ans;
queue<int> q;

void bfs()
{
    while(!q.empty())
        q.pop();
    dis[1]=0;
    EX[1]=1;
    visited[1]=1;
    q.push(1);
    while(!q.empty())
    {
        int x=q.front();
        visited[x]=1;
        q.pop();
        l.clear();
        for(int i=0;i<p[x].size();i++)
        {
            int y=p[x][i];
            if(!visited[y])
                l.push_back(y);
        }
        if(l.size()==0)
            ans+=EX[x]*dis[x];
        else
        {
            for(int i=0;i<l.size();i++)
            {
                EX[l[i]]=EX[x]/l.size();
                dis[l[i]]=dis[x]+1;
                q.push(l[i]);
            }
        }
    }
}



int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        p[x].push_back(y);
        p[y].push_back(x);
    }
    ans=0;
    met(visited,0);
    met(dis,0);
    met(EX,0);
    bfs();
    printf("%f\n",ans);
}