Lucky Coins 【HDU - 5985】【概率dp】

题目链接

题目大意

有n种硬币，每次拍一下桌子会有的硬币面向下，把朝向下的硬币拿走，接着拍桌子，直到只剩一种硬币为止，那个硬币就是幸运硬币，给你每种硬币的数量和每次拍桌这种硬币面向上的概率，问每种硬币成为幸运硬币的概率是多少

解题思路

设die[i][j] 表示第i种硬币在第j轮全部死亡的概率
那么die[i][j]=(1.0 - p[i] ^j ) ^num[i]
设li [i][j]表示这种硬币在第j轮还有剩余活着的硬币
那么li [i][j] = 1.0 - die[i][j];
所以第i种硬币是幸运硬币的概率就是到当前轮的时候，只剩第i种硬币活着其他硬币全部死亡，把这些情况加起来就是答案
注意：当前轮第i种硬币活着，第i+1轮就应该全部死亡才能是幸运硬币
因为值要求到小数点后6位，所以轮数随便走个100轮精度就应该够了
注意特判n=1的情况

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;

const int N=15;
const int M=105;
double pi[N],die[N][M],li[N][M],ans[N];
int num[N];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(ans,0,sizeof(ans));
        memset(die,0,sizeof(die));
        memset(li,0,sizeof(li));
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d %lf",&num[i],&pi[i]);
        if(n==1)
        {
            printf("1.000000\n");
            continue;
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=M-5;j++)
            {
                double x=pow(pi[i],j);
                die[i][j]=pow(1.0-x,num[i]);
                li[i][j]=1.0-die[i][j];
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<M-5;j++)
            {
                double x=li[i][j]-li[i][j+1];
                double tmp=1;
                for(int k=1;k<=n;k++)
                {
                    if(k==i)
                        continue;
                    tmp*=die[k][j];
                }
                ans[i]+=tmp*x;
            }
        }
        for(int i=1;i<=n;i++)
        {
            printf("%.6f%c",ans[i],i==n?'\n':' ');
        }
    }
    return 0;
}