【JZOJ 3072】掷骰子

Description

太郎和一只免子正在玩一个掷骰子游戏。有一个有N个格子的长条棋盘，太郎和兔子轮流掷一个有M面的骰子，骰子M面分别是1到M的数字.且掷到任意一面的概率是相同的.掷到几.就往前走几步.当谁走到第N格时，谁就获胜了。游戏中还有一个规则“反弹”.就是当一位选手要走到第N格外时.他就会后退(就像飞行棋进营一样)。

假设现在一位追手在A格.当他掷出B时：


1.A+B<N，走到第A+B.络，


2.A+B=N，走到第N格，获胜。


3.A+B≥N，走到第(N-(A+B-N)格


现在太郎和兔子分别在第x和y格.接下来是太郎掷骰子，太郎想知道他赢得比赛的概率就多少。

Solution

用DP，
因为正着推不好玩，所以考虑一下倒推，
设$f_{i,j}$表示两人分别在i和j，到i走，从终点开始倒推，

很显然，如果i=n，f直接加1，j=n直接跳过；
再分类讨论：

1、两人都在[n-m+1,n]这个区间内，i赢的概率是：

$$\frac{1}{m}+(\frac{m-1}{m})^2*\frac{1}{m}+(\frac{m-1}{m})^4*\frac{1}{m}+(\frac{m-1}{m})^6*\frac{1}{m}...$$
i一次跳到了终点，或者第一次没中，j也没中，i才中，以此类推，

$$=\frac{1}{m}\sum_{i=0}^{\infty}(\frac{m-1}{m})^{2i}$$
用等比数列表示：
$$=\frac{1}{m}*\frac{(\frac{m-1}{m})^\infty-1}{\frac{1}{m^2}-\frac{2}{m}}$$
由于 $(\dfrac{m-1}{m})^\infty$无限接近0，也就是等于0，所以
$$=\frac{1}{m}*\frac{m^2}{2m-1}=\frac{m}{2m-1}$$
所以：
$$f_{i,j}=\frac{m}{2m-1}$$

2、i在[n-m+1,n]这个区间内:

$$f_{i,j}=\dfrac{1}{m}+\dfrac{\sum_{k=j+1}^{j+m} (m-1)f_{i,k}}{m^2}$$
有可能直接跳到终点，或者从其他点跳来，对于每个的k，都有(m-1)个非终点跳到i，同时每个k刚好跳到j和每个非终点的跳到i概率积为 $m^2$；

3、j在[n-m+1,n]这个区间内:

$$f_{i,j}=\dfrac{(i+n=m?1:0)}{m^2}+\dfrac{\sum_{k=i+1}^{i+m} (m-1)f_{k,j}}{m^2}$$
如果i在n-m这个位置，那么就有 $\dfrac{1}{m}$的概率直接跳到终点，但也有可能在跳之前j先到了终点，所以要减 $\dfrac{m-1}{m}*\dfrac{1}{m}$等于 $\dfrac{1}{m^2}$，
后面的与前面的差不多，这里就不多写了；

4、i,j都不在区间内：

这个简单嘛，直接把他们后面的概率加起来除$m^2$即可，
$f_{i,j}=\dfrac{\sum_{k=i+1}^{i+m}\sum_{l=j+1}^{j+m}f_{j,k}}{m^2}$

---

我们发现每个$\sum$可以写一个矩阵后缀和来优化，
答案就是$f_{x,y}$。

复杂度：$O(n^2)$

Code

成就：代码写进700！

#include<cstdio>
#include<cstdlib>
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef double db;
const int N=2050;
int m,n,x,y;
db f[N][N];
db S(int q,int w,int q1,int w1){return f[q][w]-f[q1][w]-f[q][w1]+f[q1][w1];}
int main()
{
    scanf("%d%d%d%d",&n,&m,&x,&y);
    db q=(1.0*m/(2*m-1));
    fod(i,n,1)
        fod(j,n,1)
        {
            f[i][j]=f[i+1][j]+f[i][j+1]-f[i+1][j+1];
            if(i==n){f[i][j]++;continue;}
            if(j==n)continue;
            if(i+m>n&&j+m>n)f[i][j]+=q;
                else if(i+m>n)f[i][j]+=((m-1)*S(i,j+1,i+1,j+m+1)+m)/(m*m);
                    else if(j+m>n)f[i][j]+=((m-1)*S(i+1,j,i+m+1,j+1)+(i+m==n))/(m*m);
                        else f[i][j]+=S(i+1,j+1,i+m+1,j+m+1)/(m*m);
        }
    printf("%.6lf\n",S(x,y,x+1,y+1));
    return 0;
}