Codeforces 839C Journey

最新推荐文章于 2021-08-30 19:11:34 发布

原创最新推荐文章于 2021-08-30 19:11:34 发布 · 711 阅读

0 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

143 篇文章

订阅专栏

本文提供了一种解决CodeForces平台上的C题的方法，采用广度优先搜索算法（BFS）计算图中所有节点到根节点的加权距离，并通过遍历每个节点来求得最终答案。

题目链接：

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);
}