XDOJ 1020 ACMer去刷题吧(基础dp)

题目描述

刷题是每个ACMer必由之路，已知某oj上有n个题目，第i个题目小X能做对的概率为Pi(0<=Pi<=1，1<=i<=n) 求小X至少做对k道题的概率

输入

第一行输入一个正整数t，(t<=20)，表示有t组测试样例。 第二行输入正整数n，k，（1<=n,k<=1000） 第三行输入n个小数，分别为Pi(1<=i<=n,0<=Pi<=1)，表示小X做对第i个题目的概率。

输出

输出小X至少做对k道题的概率，并换行（保留4位小数）

样例输入

2
1 1
0.5
3 2
0.3 0.2 0.1

样例输出

0.5000
0.0980

提示

题目大意：

　　这题就是说，给你一个每道题做对的概率，然后，让你求出来至少做对k道题的概率是多少.

解题思路：

　　其实不难，只要推出来dp[i][j]玩就好了，一开始推了一会，，少了个条件(Q神给窝补上去了)

　　状态：dp[i][j]表示的是有i道题，作对j道题的概率。

　　初始状态：dp[0][0] = 1,dp[0][1] = 0;

　　状态转移方程：dp[i][j] = dp[i-1][j-1]*p[i]+dp[i-1][j]*(1-p[i]);

　　　　　　　　　dp[i][0] = (1-p[i])*dp[i-1][0]

　　最后只要求出来sum+=dp[n][k]+dp[n][k+1]+dp[n][k+2]+...+dp[n][n];

代码：

# include<cstdio>
# include<iostream>
 
using namespace std;
 
# define MAX 1234
 
double p[MAX];
double dp[MAX][MAX];
int n,k;
 
int main(void)
{
    int t;scanf("%d",&t);
    while ( t-- )
    {
        scanf("%d%d",&n,&k);
        for ( int i = 1;i <= n;i++ )
        {
            scanf("%lf",&p[i]);
        }
        dp[0][0] = 1;
        dp[0][1] = 0;
        for ( int i = 1;i <= n;i++ )
        {
            dp[i][0] = (1-p[i])*dp[i-1][0];
        }
        for ( int i = 1;i <= n;i++ )
        {
            for ( int j = 1;j <= n;j++ )
            {
                dp[i][j] = (1-p[i])*dp[i-1][j]+p[i]*dp[i-1][j-1];
            }
        }
        double sum = 0;
        for ( int i = k;i <= n;i++ )
        {
            sum+=dp[n][i];
        }
       printf("%.4lf\n",sum);
 
    }
 
 
    return 0;
}