[vijos1145]小胖吃巧克力<概率dp>

题目链接:https://vijos.org/p/1145

貌似还有一个一样的题是poj1322 chocolate,两个题只是描述不一样,意思都是一样的,不贵最近貌似poj炸了,所以也没法去poj

提交一次了

 

最近在看概率dp,想入个门,然后就误入了这个打着入门题的旗号却要啥生成函数处理的题,虽然最后借助大佬提供的奇技淫巧解决了n过大的问题

 

我们定义数组f[i][j]表示取出第i个时,桌上剩下j个的概率

初始状态f[0][0]=1.0;

f[i][j]转移有两种情况,就是要么这个i配对成功,要么配对失败

配对成功:f[i][j]+=f[i-1][j+1]*(j+1)*p;就是当前第i个和之前剩下中的第j+1个配对成功,因为之前剩下j+1个,所以i可能有j+1种可能,最后乘上每种的概率p

     这种情况要注意j+1<=i-1&&j+1<=c

配对失败:f[i][j]+=f[i-1][j-1]*(c-j+1)*p;就是第i个与之前的j-1均不配对,所以就是除开j-1种之外的可能,因为一共是c种,就是c种减去j-1种可能乘上概率p

       这种情况要注意j>0

p就是1除以c

然后有两种情况就是当m>c时和m和n不是同奇同偶时就直接输出0.000.。。。。。。。。我自己之前在这输出时没注意是多组输入,所以直接return 0;然后就runtime error了

QAQ这题的生成函数(母函数)我也不懂,可以去搜索poj1322看看大佬的方法

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<queue>
 5 #include<cmath>
 6 #include<algorithm>
 7 #include<cstdlib>
 8 #define maxn 1005
 9 using namespace std;
10 
11 int n,m,c;
12 float f[maxn][105];
13 
14 int main(){
15     while(scanf("%d",&c)!=EOF){
16         if(c==0)return 0;
17         scanf("%d%d",&n,&m);
18         memset(f,0.0,sizeof(f));
19         if(m>c||((n&1)!=(m&1))){printf("0.000
");continue;}
20         if(n>1000){if(n&1)n=1001;else n=1000;}
21         f[0][0]=1.0;
22         float p=1.0/c;
23         for(int i=1;i<=n;i++){
24             for(int j=0;j<=i&&j<=c;j++){
25                 if(j>0)f[i][j]+=f[i-1][j-1]*(c-j+1)*p;
26                 if(j+1<=i-1&&j+1<=c)
27                     f[i][j]+=f[i-1][j+1]*(j+1)*p;
28             }
29         }
30         printf("%.3f
",f[n][m]);        
31     }
32 
33 }
View Code

 

原文地址:https://www.cnblogs.com/Danzel-Aria233/p/7697715.html