2020牛客寒假算法基础集训营2 C.算概率

C.算概率

题目链接-算概率


解题思路

比赛的时候没有想到状态转移方程，太菜了

概率DP，dp[i][j]表示前i道题做对j道的概率
由于i-1时对了j题，所以第i题做错了；
由于i-1时对了j-1题，所以第i题对了；
所以可得到状态转移方程：
dp[i][j] = dp[i-1][j] * p[i] + dp[i-1][j] * (1-p[i])
注意取模为负的情况,所以可将(1-p[i])换成（M+1-p[i]）

附上代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int INF=0x3f3f3f3f;
const int N=1e5+5;
const int M=1e9+7;
typedef long long ll;
typedef pair<int,int> PII;
ll p[2010],dp[2010][2010];
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	
	ll n;
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>p[i];
	dp[0][0]=1;
	for(int i=1;i<=n;i++){
		dp[i][0]=dp[i-1][0]*(M+1-p[i])%M;
		for(int j=1;j<=i;j++)
			dp[i][j]=(dp[i-1][j]*(M+1-p[i])+dp[i-1][j-1]*p[i])%M;
	}
	for(int i=0;i<=n;i++)
		cout<<dp[n][i]<<" ";
	return 0;
}