acwing 217. 绿豆蛙的归宿（概率dp入门）

link

#include <bits/stdc++.h>
using namespace  std;
#define  int long long
//typedef long long ll;
typedef pair<int,int> pii;
#define x first 
#define y second
#define pb  push_back
#define inf 1e18
#define IOS   std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define  fer(i,a,b)  for(int i=a;i<=b;i++)
#define  der(i,a,b)  for(int i=a;i>=b;i--)
#define  me(x,val) memset(x,val,sizeof(x)) 
const int maxn=1e5+10;
const int mod=1e9+7;
int qmi(int a,int b)
{int res=1;  while(b){  if(b&1) res=res*a%mod;  a=a*a%mod;  b>>=1; } return res;}
const int N=2e6+10;
const int M=1e6+10;
int dr[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
int n,k;
int m;
int du[N];
int w[M];
int cnt;
int e[N],h[N],ne[N];
double f[N];
void add(int a,int b,int c)
{
	e[cnt]=b,w[cnt]=c; ne[cnt]=h[a],h[a]=cnt++;
}
double dp(int u)
{
	if(f[u]>=0)return f[u];
	f[u]=0;
	for(int i=h[u];~i;i=ne[i])
	{
		int v=e[i];
		f[u]+=(w[i]+dp(v))/du[u];
	}
	return f[u];
}
void solve()
{
   cin >> n >> m;
    memset(h,-1,sizeof h);
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin >> a >> b >> c;
        add(a,b,c);
        du[a]++;
    }
    memset(f,-1,sizeof f);
    printf("%.2lf\n",dp(1));
}

signed main()
{
//	IOS;
	int _=1;
	//cin>>_;
	while(_--) solve();
	return 0;
}