ABC189 F - Sugoroku2（期望dp）

最新推荐文章于 2022-10-30 10:58:12 发布

live4m

最新推荐文章于 2022-10-30 10:58:12 发布

阅读量169

点赞数

本文链接：https://blog.csdn.net/weixin_44178736/article/details/113374709

版权

本文解析了一个涉及动态规划的数学问题，通过设定期望次数函数f(x)，讨论了如何根据标记点和递推关系f(x)=k*f(0)+b来求解从点x到终点的期望次数，特别关注了f(0)的求解过程。代码实例展示了如何利用双数组和辅助变量维护k和b来求解该问题。

摘要由CSDN通过智能技术生成

题意：

在这里插入图片描述

解法：

$设f(x)为从点x到达终点的期望次数,那么:\\ 1.当x>=n时,f(x)=0.\\ 2.如果x被标记,那么f(x)=f(0).\\ 3.其他情况f(x)=\frac{f(x+1)+f(x+2)...f(x+m)}{m}+1.\\ 因为有f(0)存在,那么一定是f(x)=kf(0)+b的形式,\\ 对于每个位置,维护一下k和b即可.\\ 最后f(0)=kf(0)+b,如果k=1,那么无解,\\ 否则(1-k)f(0)=b,可解得f(0)=\frac{b}{1-k}.$

code：

#include <bits/stdc++.h>
using namespace std;
const int maxm=2e6+5;
double sumK[maxm];
double sumB[maxm];
double K[maxm];
double B[maxm];
int mark[maxm];
int n,m,k;
signed main(){
    ios::sync_with_stdio(0);
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++){
        int x;cin>>x;
        mark[x]=1;
    }
    K[n]=0,B[n]=0;
    for(int i=n-1;i>=0;i--){
        int l=i+1,r=i+m;
        if(mark[i]){
            K[i]=1,B[i]=0;
        }else{
            K[i]=(sumK[l]-sumK[r+1])/m;
            B[i]=(sumB[l]-sumB[r+1])/m+1;
        }
        sumK[i]=sumK[i+1]+K[i];
        sumB[i]=sumB[i+1]+B[i];
    }
    if(K[0]==1){
        cout<<-1<<endl;
        return 0;
    }
    double ans=B[0]/(1-K[0]);
    printf("%.10f\n",ans);
    return 0;
}