DP-牛客寒假集训营2-算概率

DP-牛客寒假集训营2-算概率

题目：

题意：

既然输入输出都是取模意义下的结果，就不管这个定义了，把 $p[i]$ 就直接理解成概率即可。

题解：

$n道题中答对i道的概率与n-1道题中答对i道题的概率是有关系的$

$考虑转移方程：dp[i][j]—i道题中答对j道题的概率，显然j>i时，dp[i][j]=0$

$设第k题答对的概率p_k，考虑前i题中，答对j题的概率的概率$

$dp[i][j]=dp[i-1][j]\ast (1-p[i])+dp[i-1][j-1]\ast p[i]$

$即为前i-1题中答对j题但第i题未答对的概率+前i-1题答对j-1道且第i题答对的概率$

$特别地，dp[0][0]=1，dp[n][0]={\prod }_{i=1}^n(1-p_i),(n>=1)。$

$类似于组合数公式C_n^m=C_{n-1}^m+C_{n-1}^{m-1}$

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=1e9+7;
const int N=2005;
int n;
ll p[N],dp[N][N];//dp[i][j]代表前i道题做对j题的概率
/*
    dp[i][j]=dp[i-1][j]*(1-p[i])+dp[i-1][j-1]*p[i]
    求dp[n][i]  i:0~n
*/
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        scanf("%lld",&p[i]);
    dp[0][0]=1,dp[0][1]=0;
    
    for(int i=1;i<=n;i++)
    {
        dp[i][0]=dp[i-1][0]*(1-p[i]+mod)%mod;//全答错
        for(int j=1;j<=i;j++)//j<=i
            dp[i][j]=((dp[i-1][j]%mod)*(1-p[i]+mod)%mod+(dp[i-1][j-1]%mod)*(p[i]%mod)%mod)%mod;
    }
    
    for(int i=0;i<=n;i++)
        printf("%lld ",dp[n][i]);
    return 0;
}