欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

C:算概率(dp)

程序员文章站 2024-02-27 13:19:45
...

传送门

题意:C:算概率(dp)

思路:

概率dp,用dp[i][j]表示做前i道题,做对j道
考虑状态转移方程:如果前i-1道题做对j道,那么第i道题应该做错了,如果前i-1道题做对j-1道,那么第i道题应该做对了

dp[i][j]=(dp[i-1][j]*(1-p[i]+mod)%mod+dp[i-1][j-1]*p[i]%mod)%mod;

代码:

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <set>
#define Pll make_pair
using namespace std;
typedef long long ll;
const int maxn=2e5+50;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const int phi=1e9+6;
const ll M=1e18+7;
int p[2050];
ll dp[2050][2050];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&p[i]);
    //dp[i][j]到第i题时,共正确j道题
    dp[0][0]=1;
    for(int i=1;i<=n;i++){
        dp[i][0]=dp[i-1][0]*(1-p[i]+mod)%mod;
        for(int j=1;j<=n;j++)
            dp[i][j]=(dp[i-1][j]*(1-p[i]+mod)%mod+dp[i-1][j-1]*p[i]%mod)%mod;
    }
    for(int i=0;i<=n;i++){
        printf("%lld ",dp[n][i]);
    }
    printf("\n");
    return 0;
}
/*

 */
相关标签: dp 牛客