【LOJ6342】 跳一跳 期望概率DP

题目描述

现有一排方块，依次编号为 1…n1\ldots n1…n。
方块 111 上有一个小人，已知当小人在方块 iii 上时，下一秒它会等概率地到方块 iii（即不动），方块 i+1i+1i+1，方块 i+2i+2i+2……方块 nnn 上。
求小人到达方块 nnn 所需要的期望时间（单位：秒）。
具体题目见：这里

题解

根据题目要求，我们可以求得递推式$f[i]=\frac{\sum\limits_{j=i+1}^{n}{f[j]}}{n-i}$
推导$:$
首先我们可以看出$f[i]=\frac{\sum\limits_{j=i}^{n}{f[j]}}{n-i+1}$
然后我们不能让$f[i]$在式子内出现，所以我们可以用条件概率$同时除以(1-\frac{1}{n-i+1})$
就可以得到$f[i]=\frac{\sum\limits_{j=i+1}^{n}{f[j]}}{n-i}$
逆元的话直接线性递推就可以

代码

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<cstdio>
const int mod=1e9+7;
const int MAXN=1e7;
int n;
int inv[MAXN];
long long dp,ans;
using namespace std;
int main()
{
    scanf("%d",&n);
    inv[1]=1;
    for(int i=2;i<=n;i++) 
        inv[i]=1ll*(mod-(mod/i))*inv[mod%i]%mod;
    for(int i=n-1;i;i--)
    {
        ans=((dp+1)*(long long )inv[n-i]+1)%mod;//DP为维护的后缀和
        dp=(dp+ans)%mod;
    }
    printf("%lld",ans);
    return 0;
}