E41 概率DP 求期望 拓扑排序

视频链接：https://www.bilibili.com/video/BV1ZT411n7AT/

Luogu P4316 绿豆蛙的归宿

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N=100010,M=2*N;
int n,m,a,b,c;
int h[N],to[M],ne[M],w[M],tot;
int out[N]; //out[u]:点u的出边条数
double f[N];

void add(int a,int b,int c){
  to[++tot]=b,w[tot]=c,
  ne[tot]=h[a],h[a]=tot;
}
double dfs(int u){
  if(f[u]) return f[u]; //记忆化搜索
  for(int i=h[u];i;i=ne[i]){
    int v=to[i]; 
    f[v]=dfs(v);
    f[u]+=(f[v]+w[i])*1/out[u];
  }
  return f[u];
}
int main(){
  scanf("%d%d",&n,&m);
  for(int i=0; i<m; i++){
    scanf("%d%d%d",&a,&b,&c);
    add(a,b,c); out[a]++;
  }
  dfs(1);
  printf("%.2lf\n",f[1]);
}

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

const int N=100005,M=2*N;
int n,m,a,b,c;
int h[N],to[M],ne[M],w[M],tot;
int in[N],k[N]; //in[x]:点x的入度
double f[N];

void add(int a,int b,int c){
  to[++tot]=b; w[tot]=c;
  ne[tot]=h[a]; h[a]=tot;
}
void topo(){
  queue<int>q; q.push(n);
  while(q.size()){
    int u=q.front(); q.pop();
    for(int i=h[u];i;i=ne[i]){
      int v=to[i];
      f[v]+=(f[u]+w[i])*1/k[v];
      if(--in[v]==0) q.push(v);
    }
  }
}
int main(){
  scanf("%d%d",&n,&m);
  for(int i=0;i<m; i++){
    scanf("%d%d%d",&a,&b,&c);
    add(b,a,c); //建反图
    in[a]++, k[a]++;    
  }
  topo();       //拓扑排序
  printf("%.2lf\n",f[1]);
}

 

练习：

HDU3853 LOOPS

HDU4035 Maze