LeetCode.837 New 21 Game

题目描述和分析

LeetCode.837的原题叙述比较繁琐，有需求的同学可以自行链接，这里给出一个简化版本。初始点数为0，每次从$[1,W]$中独立同分布地随机选取一个整数，并累积起来，直到累积和大于等于$K$为止。请问累积点数小于等于$N$的概率？（$W$，$K$，$N$都为正整数且$K<=N$）假设事件采取数字的累积和为$X$，所以根据题意所需求的概率为，

$$p(K\le X\le N)$$

举一个简单的例子，

$$Input:N=6,K=1,W=10\to p=0.6$$

这个例子求解非常简单，因为只需采样一次便可以满足大于等于$K$这一条件，故事件的概率为$p(Xwhich1\le X\le 6)=0.1$，接下来累积便可。这个例子很难说明问题的本质，编写的程序需要解决的是一个普遍问题，而不是特例。我们取个稍微大一点的$K(=3)$，并且从$X=1$开始计算概率，

$$X=1:p(a_1=1)=0.1$$

$$X=2:p(a_1=2)+p(a_1=1,a_2=1)=0.1+0.01$$

$$\begin{gathered} X=3:p(a_1=3)+p(a_1=1,a_2=2)+p(a_1=2,a_1=2) \\ +p(a_1=0.1,a_2=0.1,a_3=0.1)=0.1+0.01+0.01+0.001 \end{gathered}$$

对数列稍作转换，

$$p(X=0)=1$$

$$p(X=1)=0.1\times 1$$

$$p(X=2)=0.1\times (1+0.1)$$

$$p(X=3)=0.1\times (1+0.1+0.1\times (1+0.1))$$

便可以找到一个递推式$p(X=i)=\frac{1}{W}{\sum }_k^{i-1}p(X=k)$，当然，这个公式仅在$X\le 3$时是符合的，

$$\begin{gathered} X=4:p(a_1=4)+p(a_1=1,a_2=3)+p(a_1=2,a_2=2) \\ +p(a_1=1,a_2=1,a_3=2)=0.121 \end{gathered}$$

$$\begin{gathered} X=5:p(a_1=5)+p(a_1=1,a_2=4)+p(a_1=4,a_2=1) \\ +p(a_1=1,a_2=1,a_3=4)=0.121 \end{gathered}$$

可以看到概率值和$p(X=3)$都是一样，诸如$X=5$往后不需要添加$X=4$的情况。

还有一个点很容易被忽略，在翻看了论坛才发现（建议同学们可以上英文版LeetCode，上面的论坛比较活跃）。当在累积概率的过程中，$X>W$不能简单地从头累积，而是设定一个窗口${\sum }_{i=X-W}^{X-1}$。具体的代码如下，

代码

double new21Game(int N, int K, int W) {
    if (K == 0 || N >= K + W) 
        return 1.0;
    
    vector<double> dp(N + 1);
    dp[0] = 1.0;
    double Wsum = 1.0, res = 0.0;
    
    for (int i = 1; i <= N; ++i) {
        dp[i] = Wsum / W;
        if (i < K) Wsum += dp[i]; else res += dp[i];
        if (i - W >= 0) Wsum -= dp[i - W];
    }
    
    return res;
}

一次遍历，效率达到top级。（代码和解法是出题者在论坛中给出的，这道题非常值得琢磨）