题目链接
题目大意
有一个有向无环图,现在要从点1走到点n,第i天的花费为i,每天可以按照路走向相邻的饿节点或者保持不动,问走到点n的期望
解题思路
我们用天数期望去推花费期望
天数:me[u]=(1/(k+1))* (me[v]+1) +(1/(k+1))* (me[u]+1)
花费:dp[u]=(1/(k+1))* (me[v]+dp[v]+1)+ (1/(k+1))* (dp[u]+dp[v]+1)
其中k是和u相连的有多少个点,那么每次走向相邻点或者原地不动的概率就是(1/(k+1));
v是和u相邻的点
上面的公式化简一下就好
这个dp的顺序要用拓扑序才能保证计算的点不影响之前的点。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int N=1e5+5;
vector<int> rg[N];
int du[N],to[N],n,m;
int seq[N];
struct node
{
int v,nex;
};
node e[N*4];
int f[N],tot;
void getDAGseq(int n)
{
int top=0;
queue<int> Q;
for(int i=1;i<=n;++i){
if(!du[i])
Q.push(i);
}
while(!Q.empty())
{
int v=Q.front();
Q.pop();
seq[++top]=v;
for(int u:rg[v]){
du[u]--;
if(du[u]==0)
Q.push(u);
}
}
}
void add(int u,int v)
{
e[tot].v=v;
e[tot].nex=f[u];
f[u]=tot++;
}
void init()
{
memset(f,-1,sizeof(f));
memset(to,0,sizeof(to));
memset(du,0,sizeof(du));
for(int i=0;i<=n;i++)
rg[i].clear();
tot=0;
}
double me[N],dp[N];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
int x,y;
init();
for(int i=1;i<=m;i++)
{
scanf("%d %d",&x,&y);
rg[y].push_back(x);
du[x]++;
add(x,y);
to[x]++;
}
getDAGseq(n);//拓扑序
memset(me,0,sizeof(me));
memset(dp,0,sizeof(dp));
for(int i=2;i<=n;i++)
{
int v=seq[i];
double p=1.0/(double)(to[v]+1);
for(int j=f[v];j!=-1;j=e[j].nex)
{
int vv=e[j].v;
me[v]+=(me[vv]+1)*p;
}
me[v]+=p;
me[v]*=1.0/(1.0-p);
for(int j=f[v];j!=-1;j=e[j].nex)
{
int vv=e[j].v;
dp[v]+=(me[vv]+dp[vv]+1)*p;
}
dp[v]+=(me[v]+1)*p;
dp[v]*=1.0/(1.0-p);
//printf("%f %f\n",me[v],dp[v]);
}
printf("%.2f\n",dp[1]);
}
return 0;
}