1. 程式人生 > >Robberies (01背包dp變形)

Robberies (01背包dp變形)

答案 bsp .net cin ems i++ wid str algorithm

技術分享圖片技術分享圖片?

題意:一個強盜要搶劫銀行又不想被抓到,所以要進行概率分析求他在不被抓的情況下能搶最多的錢。他給定T(樣例個數),N(要搶的銀行的個數),P(被抓的概率要小於P)Mj(強盜能搶第j個銀行Mj元錢),Pj(強盜搶第j個銀行被抓的概率為Pj)。

思路:被抓的概率不好直接求出來,但可以直接求出不被抓的概率,則有狀態轉移方程dp[j] = max(dp[j], dp[j-b[i].money]*b[i].p)表示搶到j元錢被抓的最大的概率是多少。然後逆序遍歷第一個小於P的dp的下標就是答案。

PS:數組的下標表示的是可以搶到的錢數,所以不能按100來開數組。。。。。。。。(T到吐血)

代碼:

技術分享圖片
 1
#include <iostream> 2 #include <queue> 3 #include <cstdio> 4 #include <algorithm> 5 #include <cmath> 6 #include <cstring> 7 #include <queue> 8 #include <map> 9 #include <vector> 10 #define INF 0x3f3f3f3f 11 #define FRE() freopen("in.txt","r",stdin) 12
13 using namespace std; 14 typedef long long ll; 15 const int maxn = 10000; 16 double dp[maxn]; 17 struct Bank 18 { 19 int money; 20 double p; 21 } bb[maxn]; 22 23 int main() 24 { 25 ios::sync_with_stdio(false); 26 int T,n; 27 double p; 28 cin>>T; 29 while(T--) 30 {
31 int sum = 0; 32 cin>>p>>n; 33 for(int i = 0; i<n; i++) 34 { 35 cin>>bb[i].money>>bb[i].p; 36 sum += bb[i].money; 37 } 38 memset(dp,0,sizeof(dp)); 39 dp[0] = 1; 40 for(int i = 0; i<n; i++) 41 { 42 for(int j=sum; j>=bb[i].money; j--) 43 { 44 dp[j] = max(dp[j], dp[j-bb[i].money]*(1.0 - bb[i].p)); 45 } 46 } 47 int ans = 0; 48 for(int i = sum; i>=0; i--) 49 { 50 if(1-dp[i]<=p) 51 { 52 ans = i; 53 break; 54 } 55 } 56 cout<<ans<<endl; 57 } 58 return 0; 59 }
View Code

Robberies (01背包dp變形)