牛客寒假赛2 c 算概率(概率dp+逆元)

题目描述
牛牛刚刚考完了期末,尽管 牛牛 做答了所有 n 道题目,但他不知道有多少题是正确的。 不过,牛牛 知道第 i 道题的正确率是 pi​。 牛牛 想知道这 n 题里恰好有 0,1,…,n 题正确的概率分别是多少,对 1e9+7取模。

输入描述:
第一行,一个正整数 n。第二行,n 个整数 p1,p2,pn​,在模 1e9+7 意义下给出。保证 1≤n≤2000
输出描述:
输出一行 n+1 个用空格隔开的整数表示答案(对 1e9+7 取模)。

输入
1
500000004
输出
500000004 500000004
说明
有 1 道题,做对的概率是 1 \21​( 1 \2​ 在模 1e9+7意义下为 500000004 )。

这道题应用了逆元加概率dp,实际上逆元也不用自己求,直接用pi%1e9+7即可表示概率,逆元是因为a/b运算中,如果b过大则精度不够,我们需要转换为乘法形式,
设c是b的逆元,则有bc≡1(mod m);//表达的意义为c是b的逆元在mod m的情况下c相当于b的倒数
则(a/b)%m = (a/b)1%m = (a/b)bc%m = ac(mod m);
即a/b的模等于a
b的逆元的模;三个等号代表恒等,
本题直接把a/b的逆元告诉了我们,我们就不用去自己计算a/b的逆元直接用P[I]mod1e9来表示即可
接下来就是一个概率dp的问题,这道概率dp的状态转移可以设计为前i个体重正确数为j
得到状态转移方程dp[i][j]=dp[i-1][j-1]pi+dp[i-1][j](1-pi);
起点为dp[0][0]=1;先对每一个i的j=0的情况计算概率之后就可以递推求解了
ac代码如下

#include<bits/stdc++.h>
using namespace std;
long long int dp[2005][2005],f[2005];
int mod=1e9+7;
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    cin>>f[i];
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        dp[i][0]=dp[i-1][0]*(mod+1-f[i])%mod;//mod+1-f[i]相当于1-pi
        for(int j=1;j<=i;j++)
            dp[i][j]=(dp[i-1][j-1]*f[i]+dp[i-1][j]*(mod+1-f[i]))%mod;
    }
    for(int i=0;i<=n;i++)
    {
        printf("%d ",dp[n][i]);
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值