【题解】CH3802 绿豆蛙的归宿 拓扑排序+数学期望

题目链接

背景

随着新版百度空间的下线，Blog宠物绿豆蛙完成了它的使命，去寻找它新的归宿。

描述

给出一个有向无环的连通图，起点为1终点为N，每条边都有一个长度。绿豆蛙从起点出发，走向终点。
到达每一个顶点时，如果有K条离开该点的道路，绿豆蛙可以选择任意一条道路离开该点，并且走向每条路的概率为 1/K 。
现在绿豆蛙想知道，从起点走到终点的所经过的路径总长度期望是多少？

输入格式

第一行: 两个整数 N M，代表图中有N个点、M条边
第二行到第 1+M 行: 每行3个整数 a b c，代表从a到b有一条长度为c的有向边

输出格式

从起点到终点路径总长度的期望值，四舍五入保留两位小数。

样例输入

4 4
1 2 1
1 3 2
2 3 3
3 4 4

样例输出

7.00

数据范围与约定

对于20%的数据 N<=100
对于40%的数据 N<=1000
对于60%的数据 N<=10000
对于100%的数据 N<=100000，M<=2*N

---

把原图反着建，从终点跑一遍拓扑序

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=1e5+10;
int n,m,tot,head[N],deg[N],out[N];//deg和out都记录入度 
double dis[N];//路径长度期望 
struct node{
    int v,w,nx;
}edge[N<<1];
void addedge(int u,int v,int w)
{
    edge[tot].v=v;
    edge[tot].w=w;
    edge[tot].nx=head[u];
    head[u]=tot++;
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    memset(head,-1,sizeof(head));
    int u,v,w;
    for(int i=1;i<=m;i++)scanf("%d%d%d",&u,&v,&w),addedge(v,u,w),deg[u]++,out[u]++;
    queue<int>q;
    q.push(n);
    while(!q.empty())
    {
        u=q.front();q.pop();
        for(int i=head[u];~i;i=edge[i].nx)
        {
            int v=edge[i].v;
            dis[v]+=(dis[u]+edge[i].w)/deg[v];
            if(!(--out[v]))q.push(v);
        }
    }
    printf("%.2f\n",dis[1]);
    return 0;
}

总结

把期望结合到图论里，感觉这样的题做过几道，还有重边自环的那种……