One Person Game

题目：
https://zoj.pintia.cn/problem-sets/91827364500/problems/91827368253

有三个骰子，面值分别是$1$到$k_1$，$1$到$k_2$，$1$到$k_3$，扔到的概率等可能。每轮扔三个骰子，扔出的值之和加到$ans$上，当遇到$k1=a,k2=b,k3=c$时，$ans=0$重新开始累加，当$ans>n$时停止。问停止时轮数的期望。
$0\le n\le 500;1<K_1,K_2,K_3\le 6;1\le a\le K_1;1\le b\le K_2,1\le c\le K3$

思路：
令$f(i)$表示从分数为$i$时到停止时的期望，边界值$f(i)=0(i>n)$，$P$表示扔到$a,b,c$的概率($P=\frac{1}{k_1{k}_2{k}_3}$)，$p_j$表示扔到$j$分的概率(不包含$a,b,c$这种情况)，有
$$\begin{array}{rl}f(i)& =1+Pf(0)+\sum _{j=3}^{k_1+k_2+k_3}{p}_{j}f(i+j)(i>n时，反正f(i)=0)\end{array}$$
$f(0)$就是要求的，发现会成环，但是如果先不考虑$f(0)=1+Pf(0)+{\sum }_j{p}_{j}f(j)$，可以看成初始值为$f(0)$的一个递推数列。所以可以求出通项公式。
设$f(i)=A(i)f(0)+B(i)$(因为是线性的)，代入得
$$\begin{array}{rl}A(i)f(0)+B(i)& =1+Pf(0)+\sum _{j=3}^{k_1+k_2+k_3}{p}_j[A(i+j)f(0)+B(i+j)]\\ A(i)f(0)+B(i)& =1+f(0)(P+\sum _{j=3}^{k_1+k_2+k_3}{p}_{j}A(i+j))+\sum _j{p}_{j}B(i+j)\\ & \{\begin{array}{l}A(i)=P+{\sum }_{j=3}^{k_1+k_2+k_3}{p}_{j}A(i+j)可以递推求\\ B(i)=1+{\sum }_{j=3}^{k_1+k_2+k_3}{p}_{j}B(i+j)\end{array}\end{array}$$
再把方程$f(0)=1+Pf(0)+{\sum }_{j=3}^{k_1+k_2+k_3}{p}_{j}f(j)$考虑进去，有
$$\begin{array}{rl}f(0)& =1+Pf(0)+\sum _{j=3}^{k_1+k_2+k_3}{p}_j[A(j)f(0)+B(j)]\\ f(0)& =1+\sum _{j=3}^{k_1+k_2+k_3}{p}_{j}B(j)+(P+\sum _{j=3}^{k_1+k_2+k_3}{p}_{j}A(j))f(0)\\ f(0)& =B(0)+A(0)f(0)\end{array}$$
其实会发现$f(0)$也满足通项公式。

注意： 在算$p_j$时排除$a,b,c$的情况。

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
#define eps 1e-12
using namespace std;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,k[4],a,b,c;
        int i,j,l;
        scanf("%d%d%d%d%d%d%d",&n,&k[1],&k[2],&k[3],&a,&b,&c);
        double p[22];
        memset(p,0,sizeof(p));
        for(i=1;i<=k[1];i++) for(j=1;j<=k[2];j++) for(l=1;l<=k[3];l++) p[i+j+l]+=1.0/(k[1]*k[2]*k[3]);  //求出掷出k点的概率
        p[a+b+c]-=1.0/(k[1]*k[2]*k[3]);  //单独考虑下分别a,b,c点的情况
        p[0]=1.0/(k[1]*k[2]*k[3]);
        double A[600],B[600];
        memset(A,0,sizeof(A));
        memset(B,0,sizeof(B));
        for(i=n;i>=0;i--)           //递推求解
        {
            A[i]=p[0];
            B[i]=1;
            if(i==n) continue;
            for(j=1;j<=k[1]+k[2]+k[3];j++)
            {
                A[i]+=p[j]*A[i+j];
                B[i]+=p[j]*B[i+j];
            }
        }
        printf("%.15f\n",B[0]/(1-A[0]));
    }
    return 0;
}