LightOJ 1079 Just another Robbery 概率DP/01背包

Just another Robbery LightOJ - 1079

抢 $n$ 个银行，每个银行被抓的概率分别是 $p[i]$，每个银行各有 $m[i]$ 块钱，能承受的最大被抓概率是 $P$，问最多能抢多少钱？

转化一下，相当于要保证安全的概率不低于 $1-P$，抢各个银行对应的安全概率为 $1-p[i]$，要保证抢的银行的安全概率连乘起来不低于 $1-P$，典型 $01$ 背包。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
//#define WINE
#define MAXN 110
#define MAXM 10010
using namespace std;
int T,sum,m[MAXN],res,n,iCase;
double P,p[MAXN],dp[MAXM];
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d",&T);
    while(T--){
        scanf("%lf %d",&P,&n);sum=0;P=1-P;
        for(int i=1;i<=n;i++){
            scanf("%d %lf",&m[i],&p[i]);
            p[i]=1-p[i];
            sum+=m[i];
        }
        memset(dp,0,sizeof(dp));
        dp[0]=1;
        for(int i=1;i<=n;i++)
            for(int j=sum;j>=m[i];j--)
                dp[j]=max(dp[j],dp[j-m[i]]*p[i]);
        for(res=sum;res>=0;res--)
            if(dp[res]>P)
                break;
        printf("Case %d: %d\n",++iCase,res);
    }
    return 0;
}