HDU 2955 Robberies(01背包问题)

来源:互联网 发布:配音软件手机软件 编辑:程序博客网 时间:2024/04/29 22:07

Description
一个劫匪要去抢劫n家银行,每家银行有一定的现金,每抢一家银行该劫匪有一定几率被警察抓住,但是当该劫匪被抓住的几率不大于p时他可以逃脱,问该劫匪在不被捕的情况下最多能抢到多少钱?
Input
第一行为用例组数T,每组用例第一行为一个浮点数P和一个整数n分别表示被捕的几率上限以及该劫匪计划抢劫的银行数量,之后n行每个一个整数v和一个浮点数p表示该家银行的现金数以及该劫匪抢劫该家银行被捕的几率
Output
对于每组用例,输出该劫匪在不被捕的情况最多能抢到多少钱
Sample Input
3
0.04 3
1 0.02
2 0.03
3 0.05
0.06 3
2 0.03
2 0.03
3 0.05
0.10 3
1 0.03
2 0.02
3 0.05
Sample Output
2
4
6
Solution
将被捕几率看作重量,问题转化为01背包问题,dp[i]表示抢钱数为i时不被捕的最大几率,转移方程dp[j]=max(dp[j],dp[j-v[i]]*(1-p[i])),初始化dp[0]=1,最后从sum(总钱数)到0循环钱数找到符合dp[i]<1-p的最大i值即为答案
Code

#include<stdio.h>#include<string.h>#define maxn 11111#define max(x,y) ((x)>(y)?(x):(y))double p,dp[maxn];int n,sum;struct node{    int v;    double p;}m[maxn];int main(){    int t;    scanf("%d",&t);    while(t--)    {        sum=0;        scanf("%lf%d",&p,&n);        for(int i=0;i<n;i++)            {            scanf("%d%lf",&m[i].v,&m[i].p);            sum+=m[i].v;        }        memset(dp,0,sizeof(dp));        dp[0]=1;//一毛钱不抢肯定不会被捕         for(int i=0;i<n;i++)//01背包             for(int j=sum;j>=m[i].v;j--)                dp[j]=max(dp[j],dp[j-m[i].v]*(1.0-m[i].p));        for(int i=sum;i>=0;i--)//找到不被捕情况下抢钱最大值             if(dp[i]>1.0-p)            {                printf("%d\n",i);                break;            }    }    return 0;}
0 0
原创粉丝点击