CodeForces540 D. Bad Luck Island (概率dp)

题意：

岛上有三个物种，石头、剪刀、布
其中石头有r个，剪刀有s个，布有p个

三个物种之间存在克制关系：石头克制剪刀，剪刀克制布，布克制石头
当两个不同的物种相遇时，被克制的物种会被杀掉

现在要计算出：在较长一段时间之后，只剩下石头/剪刀/布的概率分别是多少

数据范围：r,s,p<=100

解法：

令d[i][j][k]表示石头i个,剪刀j个,布k个的概率

一开始d[s][r][p]=1,然后考虑接下来如何转移:
当前克制关系的对数sum=s*r+r*p+p*s
s克制r,d[s][r-1][p]+=(s*r)/sum*d[s][r][p]
r克制p,d[s][r][p-1]+=(r*p)/sum*d[s][r][p]
p克制s,d[s-1][r][p]+=(p*s)/sum*d[s][r][p]
倒着dp就行了

最后对d[i][0][0]求和就是仅存石头的答案,其他同理

code：

#include<bits/stdc++.h>
using namespace std;
const int maxm=105;
double d[maxm][maxm][maxm];
int r,s,p;
signed main(){
    cin>>r>>s>>p;
    d[r][s][p]=1;
    for(int i=r;i>=0;i--){
        for(int j=s;j>=0;j--){
            for(int k=p;k>=0;k--){
                if(i*j+j*k+k*i==0)continue;//防止出现除0
                if(i>=1){
                    d[i-1][j][k]+=1.0*k*i/(i*j+j*k+k*i)*d[i][j][k];
                }
                if(j>=1){
                    d[i][j-1][k]+=1.0*i*j/(i*j+j*k+k*i)*d[i][j][k];
                }
                if(k>=1){
                    d[i][j][k-1]+=1.0*j*k/(i*j+j*k+k*i)*d[i][j][k];
                }
            }
        }
    }
    double ans[3]={0};
    for(int i=1;i<=r;i++){
        ans[0]+=d[i][0][0];
    }
    for(int j=1;j<=s;j++){
        ans[1]+=d[0][j][0];
    }
    for(int k=1;k<=p;k++){
        ans[2]+=d[0][0][k];
    }
    for(int i=0;i<3;i++){
        printf("%.10f ",ans[i]);
    }
    return 0;
}

---