CodeForces 601 C.Kleofáš and the n-thlon(概率DP)

Description
m个人参加n场比赛,每场比赛这m个人的排名都是一个1~m的排列,且各种得分情况概率相同,现在某个人知道其n场比赛的排名,问其最终排名的期望
Input
第一行两整数n和m表示比赛场数和参赛人数,之后n行第i行一整数a[i]表示该人第i场比赛的排名(1<=n<=100,1<=m<=1000)
Output
输出该人n场比赛后的总排名的期望值
Sample Input
4 10
2
1
2
1
Sample Output
1.0000000000000000
Solution
设s=a[1]+a[2]+…+a[n],想知道这个人的最终排名,只需要知道n场比赛后另外m-1个人中总排名低于s的期望人数即可,因为每场比赛所有排名情况出现概率相同,所以另外m-1个人都是一样的,算一个人即可,以dp[i][j]表示i场比赛后这m-1个中总排名是j的期望人数,dp[0][0]=m-1表示还没比赛时排名是0的期望人数就是这m-1个人,在第i场比赛这m-1个人的排名都等概率取1~m中除a[i]的m-1个数,故有转移方程dp[i][j]=sum{dp[i-1][j-k],1<=k<=m,k!=a[i]}/(m-1),用前缀和优化下该转移,dp[i-1][k]对区间[dp[i][k+1],dp[i][k+m]]扣去dp[i][k+a[i]]点的贡献都是dp[i-1][k]/(m-1),所以每次在端点处更新最后来一遍前缀和即完成一步转移,最终答案就是dp[n][1]+…+dp[n][s-1]+1,因为该人的排名要在这些总排名靠前的人后面一名
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 101
int n,m,s,a[maxn];
double dp[111111],dpp[111111];
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        s=0;
        for(int i=1;i<=n;i++)scanf("%d",&a[i]),s+=a[i];
        if(m==1)
        {
            printf("1\n");
            continue;
        }
        memset(dp,0,sizeof(dp));
        dp[0]=m-1;
        for(int i=1;i<=n;i++)
        {
            memset(dpp,0,sizeof(dpp));
            for(int j=0;j<=n*m;j++)
            {
                double p=dp[j]/(m-1);
                dpp[j+1]+=p,dpp[j+m+1]-=p;
                dpp[j+a[i]]-=p,dpp[j+a[i]+1]+=p;
            }
            dp[0]=dpp[0];
            for(int j=1;j<=n*m;j++)dpp[j]+=dpp[j-1],dp[j]=dpp[j];
        }
        double ans=0;
        for(int j=0;j<s;j++)ans+=dp[j];
        printf("%.12f\n",ans+1.0);
    }
    return 0;
}                                              
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值