题目描述
牛牛刚刚考完了期末,尽管 牛牛 做答了所有
道题目,但他不知道有多少题是正确的。
不过,牛牛 知道第
道题的正确率是
。
牛牛 想知道这
题里恰好有
题正确的概率分别是多少,对
取模。
对
取模的含义是:对于一个
的不可约分数 $a\over $,存在
使得
,
即为
对
取模的结果。
输入描述:
第一行,一个正整数
。
第二行,
个整数
,在模
意义下给出。
保证
。
输出描述:
输出一行 个用空格隔开的整数表示答案(对 取模)。
输入
1
500000004
输出
500000004 500000004
说明
有 1 道题,做对的概率是 ( )。
题解
- 定义
- 转移时考虑第j道题是否做对,转移方程为: 。
- 时间复杂度 。
AC-Code
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 2005 + 7;
const int mod = 1e9 + 7;
ll dp[maxn][maxn]; // 前i道题作对j道的概率
ll a[maxn];
int main() {
int n; cin >> n;
for (int i = 1; i <= n; ++i) cin >> a[i];
for (int i = dp[0][0] = 1; i <= n; ++i) {
dp[i][0] = dp[i - 1][0] * (mod + 1 - a[i]) % mod;
for (int j = 0; j <= n; ++j) {
dp[i][j] = (dp[i - 1][j] * (mod + 1 - a[i]) + dp[i - 1][j - 1] * a[i]) % mod;
}
}
for (int i = 0; i <= n; ++i)
cout << dp[n][i] << " ";
}