UVA11021 Tribles 概率dp

时间:2023-03-09 08:18:33
UVA11021 Tribles     概率dp

题目传送门

题意:开始有$k$只兔子,每只都是活一天就死,每只死前都会有$pi$的概率生出$i$只兔子。求$m$天后兔子死光的概率。

思路: 

  设$f[i]$为一只兔子在第i天死完的概率,那么答案就是$f[m]^k$。

  所以关键是求$f[i]$.

     由全概率公式得到

    $f[i]=p0+p1*f[i-1]+p2*f[i-1]^2+...+pn*f[i-1]^n$

  这个式子要怎么理解呢?p0是一只兔子第一天就死完的概率。p1是一只兔子在第一天生出了一只兔子,那么这种情况下在第i天死完的概率就是p1*f[i-1],由于兔子死亡是独立重复时间,所以概率以指数的形式相乘。

  

#pragma GCC optimize (2)
#pragma G++ optimize (2)
#pragma comment(linker, "/STACK:102400000,102400000")
#include<bits/stdc++.h>
#include<unordered_map>
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define dep(i,b,a) for(int i=b;i>=a;--i)
#define clr(a,b) memset(a,b,sizeof(a))
#define pb push_back
#define pii pair<int,int >
using namespace std;
typedef long long ll;
ll rd()
{
ll x=,f=;char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
return x*f;
}
const int maxn=;
const int inf=0x3f3f3f3f;
int n,m,k,T;
double dp[maxn],f[maxn],p[maxn];
int main(){
cin>>T;
int cat=;
while(T--){
cin>>n>>k>>m;
double res=;
rep(i,,n-){
scanf("%lf",&p[i]);
}
f[]=p[];
rep(i,,m){
f[i]=;
rep(j,,n-){
f[i]+=p[j]*pow(f[i-],j);
}
}
printf("Case #%d: %.7f\n",cat++,pow(f[m],k));
}
}