DP-牛客寒假集训营2-算概率
题目:
题意:
既然输入输出都是取模意义下的结果,就不管这个定义了,把 p [ i ] p[i] p[i] 就直接理解成概率即可。
题解:
n 道 题 中 答 对 i 道 的 概 率 与 n − 1 道 题 中 答 对 i 道 题 的 概 率 是 有 关 系 的 n道题中答对i道的概率与n-1道题中答对i道题的概率是有关系的 n道题中答对i道的概率与n−1道题中答对i道题的概率是有关系的
考 虑 转 移 方 程 : d p [ i ] [ j ] — i 道 题 中 答 对 j 道 题 的 概 率 , 显 然 j > i 时 , d p [ i ] [ j ] = 0 考虑转移方程:dp[i][j]—i道题中答对j道题的概率,显然j>i时,dp[i][j]=0 考虑转移方程:dp[i][j]—i道题中答对j道题的概率,显然j>i时,dp[i][j]=0
设 第 k 题 答 对 的 概 率 p k , 考 虑 前 i 题 中 , 答 对 j 题 的 概 率 的 概 率 设第k题答对的概率\ p_k,考虑前i题中,答对j题的概率的概率 设第k题答对的概率 pk,考虑前i题中,答对j题的概率的概率
d p [ i ] [ j ] = d p [ i − 1 ] [ j ] ∗ ( 1 − p [ i ] ) + d p [ i − 1 ] [ j − 1 ] ∗ p [ i ] dp[i][j]=dp[i-1][j]*(1-p[i])+dp[i-1][j-1]*p[i] dp[i][j]=dp[i−1][j]∗(1−p[i])+dp[i−1][j−1]∗p[i]
即 为 前 i − 1 题 中 答 对 j 题 但 第 i 题 未 答 对 的 概 率 + 前 i − 1 题 答 对 j − 1 道 且 第 i 题 答 对 的 概 率 即为前i-1题中答对j题但第i题未答对的概率+前i-1题答对j-1道且第i题答对的概率 即为前i−1题中答对j题但第i题未答对的概率+前i−1题答对j−1道且第i题答对的概率
特 别 地 , d p [ 0 ] [ 0 ] = 1 , d p [ n ] [ 0 ] = ∏ i = 1 n ( 1 − p i ) , ( n > = 1 ) 。 特别地,dp[0][0]=1,dp[n][0]=\prod_{i=1}^{n} ({1-p_i}) ,(n>=1)。 特别地,dp[0][0]=1,dp[n][0]=∏i=1n(1−pi),(n>=1)。
类 似 于 组 合 数 公 式 C n m = C n − 1 m + C n − 1 m − 1 类似于组合数公式\ C_n^m=C_{n-1}^m+C_{n-1}^{m-1} 类似于组合数公式 Cnm=Cn−1m+Cn−1m−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;
}