dp+逆元+概率 牛客-算概率

 

题目描述

牛牛刚刚考完了期末，尽管 牛牛 做答了所有 n\text{}nn 道题目，但他不知道有多少题是正确的。
不过，牛牛 知道第 i\text{}ii 道题的正确率是 pip_ipi​。
牛牛 想知道这 n 题里恰好有 0,1,…,n0,1,\dots,n0,1,…,n 题正确的概率分别是多少，对 109+710^9+7109+7 取模。
对 109+710^9+7109+7 取模的含义是：对于一个 b≠0b\neq 0b​=0 的不可约分数 aba\over bba​，存在 q\text{}qq 使得 b×q mod (109+7)=ab\times q \bmod (10^9+7) =ab×qmod(109+7)=a，q\text{}qq 即为 aba\over bba​ 对 109+710^9+7109+7 取模的结果。

输入描述:

第一行，一个正整数 n\text{}nn 。
第二行，n\text{}nn 个整数 p1,p2,…,pnp_1,p_2,\dots,p_np1​,p2​,…,pn​，在模 109+710^9+7109+7 意义下给出。
保证 1≤n≤20001\leq n\leq 20001≤n≤2000。

输出描述:

输出一行 n+1\text{}n+1n+1 个用空格隔开的整数表示答案（对 109+710^9+7109+7 取模）。

示例1

输入

复制1 500000004

1
500000004

输出

复制500000004 500000004

500000004 500000004

说明

有 1 道题，做对的概率是 121 \over 221​ （ 121 \over 221​ 在模 109+710^9+7109+7 意义下为 500000004 ）。

题目链接 ：

https://ac.nowcoder.com/acm/contest/3003/C

先要了解逆元表示的概率，可以看我之前的博客，根据费马小定理可以推出组合数的简便算法 上边写的有逆元标识概率，然后这一题题意说的是，给你做对一题到n道题的概率，让你算做对0到n道题的概率，这里用到了一个dp。

先设定一个数组pp【i】【j】表示前i道题中做对j道题的个数，那么递推的表达式可以写成

pp【i】【j】=pp【i-1】【j】*（mod+1-p【i】）+pp【i-1】【j-1】*p【i】；

AC的代码如下

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int mod=1e9+7;
ll p[2000+10],pp[2000+10][2000+10];
int main()
{
	ll n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>p[i];
	}
	pp[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<=i;j++)
		{
			pp[i][j]=(pp[i-1][j]*(mod+1-p[i])+pp[i-1][j-1]*p[i])%mod;
		}
	}
	for (int i = 0; i <= n; ++i)
    cout << pp[n][i] << ' ';
  return 0;
}