Lightoj 1265(概率dp)

链接：点击打开链接

题意：你和n个老虎，m个鹿，被困在一座岛上，每两个物种随机相遇，老虎和老虎相遇则会减少两个老虎，老虎和鹿相遇则会减少一个鹿，人和老虎相遇则人直接死亡，人和鹿相遇，可以自行选择是否杀鹿，问最后没有老虎并且人成功存活的概率

代码：

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
double dp[1005][1005];
int main(){
    double ans;
    int t,n,m,i,j,cas,sum;
    scanf("%d",&t);                             //dp[i][j]表示剩i个老虎，j个鹿的概率
    for(cas=1;cas<=t;cas++){
        scanf("%d%d",&n,&m);
        for(i=0;i<=n;i++)
        for(j=0;j<=m;j++)
        dp[i][j]=0;
        dp[n][m]=1;
        for(i=n;i>=0;i--){
            for(j=m;j>=0;j--){
                if((i==0))
                continue;
                sum=i*(i-1)/2+i*j+i+j;          //所有情况的总数，老虎和老虎，老虎和鹿，人和老虎，人和鹿
                if(i>=2)
                dp[i-2][j]=(dp[i-2][j]+dp[i][j]*(i*(i-1)/2)*1.0/sum);
                if(i>=1&&j>=1)
                dp[i][j-1]=(dp[i][j-1]+dp[i][j]*(i*j*1.0/sum));
                if(j>=1)                        //减少老虎和鹿的所有情况的概率
                dp[i][j-1]=(dp[i][j-1]+dp[i][j]*j*1.0/sum);
            }
        }
        ans=0;
        for(j=0;j<=m;j++)               
        ans+=dp[0][j];
        printf("Case %d: %.7lf\n",cas,ans);
    }
    return 0;
}