code(vs)2488 绿豆蛙的归宿(拓扑排序+动规)

最新推荐文章于 2023-08-10 08:53:12 发布

Koakuma丶珏

最新推荐文章于 2023-08-10 08:53:12 发布

阅读量356

点赞数 1

分类专栏： # 动态规划文章标签： dp 动态规划拓扑排序 codevs2488

本文链接：https://blog.csdn.net/Aiwen1413/article/details/53697557

版权

动态规划专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题意：就是给一个终点和一个起点，有一只青蛙，从起点往终点走，求其到终点所走长度的期望值。

题解：首先拿到题目以后，想到的是拓扑排序，然后加上dp，dp[i]表示i点到终点的期望值，那dp[v] = (dp[u]+node[i].v)/to[ v], node[i].v表示这条边的出度，to[v]表示v的出度。这样就得到了dp的状态转移方程，为了方便，我采用反向存图。

下面附上代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
const int MAXN = 200005;
int head[MAXN] = {0}, next[MAXN] = {0}, tot = 0;
struct Node
{
    int to;
    double v;
}node[MAXN];
void build(int from, int to, double v)
{
    node[++tot].to = to;
    node[tot].v = v;
    next[tot] = head[from];
    head[from] = tot;
}

double dp[MAXN];
int n, m;
int in1[MAXN] = {0}, in2[MAXN] = {0};

queue<int> q;
void toposort()
{
    for(int i = 1; i <= n; i++)
        if(!in1[i]) q.push(i);
    while(!q.empty())
    {
       int x= q.front();
       q.pop();
       for(int i = head[x]; i; i = next[i])
       {
           int v= node[i].to;
           dp[v] += (dp[x]+node[i].v)*1.0/in2[v];//状态转移方程
           if(!--in1[v]) q.push(v);
       }
    }
}

int main()
{
   scanf("%d%d", &n, &m);
   int from, to, v;
   for(int i = 0; i < m; i++)
   {
       scanf("%d%d%d", &from, &to, &v);
       build(to, from, v); // 方向建图
       in1[from]++;
       in2[from]++;
   }
   toposort();
   printf("%.2lf\n", dp[1]);
}