POJ 2151 Check the difficulty of problems

1368 단어 DP 확률
기본적으로 DP죠. 통계적으로 0이 안 터질 확률을 n문제보다 작고 0이 안 터질 확률을 줄이면 물 건너갈 수 있어요...
이 문제는 경계가 좀 많을 수 있으니 주의해서 쓰시오
#include<cstdio>
#include<cstring>
using namespace std;

#define T 1010
#define M 40

int m,t,n;
double p[T][M];
double dp[T][M][M];
double sum[T][M];

int main(){
    int i,j,k;
    while(scanf("%d %d %d",&m,&t,&n)!=EOF&&m+n+t){
        for(i=1;i<=t;i++){
            for(j=1;j<=m;j++){
                scanf("%lf",&p[i][j]);
            }
        }
        memset(dp,0,sizeof(dp));
        memset(sum,0,sizeof(sum));
        for(k=1;k<=t;k++){
            dp[k][0][0]=1;
            for(i=1;i<=m;i++){
                dp[k][i][0]=dp[k][i-1][0]*(1-p[k][i]);
                for(j=1;j<i;j++){
                    dp[k][i][j]=dp[k][i-1][j]*(1-p[k][i])+dp[k][i-1][j-1]*p[k][i];
                }
                dp[k][i][i]=dp[k][i-1][i-1]*p[k][i];
            }
            sum[k][0]=dp[k][m][0];
            for(i=1;i<=m;i++) sum[k][i]=sum[k][i-1]+dp[k][m][i];
        }
        double p1,p2;
        p1=1;
        for(i=1;i<=t;i++){
            p1*=1-dp[i][m][0];
        }
        p2=1;
        for(i=1;i<=t;i++){
            p2*=sum[i][n-1]-sum[i][0];
        }
        double ret=p1-p2;
        printf("%.3f
",ret); } return 0; }

좋은 웹페이지 즐겨찾기