Leetcode 837.新21点（New 21 Game）

Leetcode 837.新21点

1 题目描述（Leetcode题目链接）

  爱丽丝参与一个大致基于纸牌游戏 “21点” 规则的游戏，描述如下：

爱丽丝以 0 分开始，并在她的得分少于 K 分时抽取数字。 抽取时，她从 [1, W] 的范围中随机获得一个整数作为分数进行累计，其中 W 是整数。 每次抽取都是独立的，其结果具有相同的概率。

当爱丽丝获得不少于 K 分时，她就停止抽取数字。 爱丽丝的分数不超过 N 的概率是多少？

输入：N = 10, K = 1, W = 10
输出：1.00000
说明：爱丽丝得到一张卡，然后停止。

输入：N = 6, K = 1, W = 10
输出：0.60000
说明：爱丽丝得到一张卡，然后停止。
在 W = 10 的 6 种可能下，她的得分不超过 N = 6 分。

提示：

• 0 <= K <= N <= 10000
• 1 <= W <= 10000
• 如果答案与正确答案的误差不超过 10^-5，则该答案将被视为正确答案通过。
• 此问题的判断限制时间已经减少。

2 题解

  定义$dp[i]$：当前得分为$i$时，爱丽丝分数不超过$N$的概率。这样定义的话，本题的答案就是$dp[0]$。

  首先明确，爱丽丝最高可获得的分数为$K-1+W$。因为一旦爱丽丝获得超过$K$分时，就会停止拿牌，当爱丽丝手中得分为$K-1$时，再拿牌可能拿到的最大值就是$K-1+W$，因此$dp$数组长度就应该是$K+W$。

  再来看初始化，根据$dp[i]$的定义有:
$$dp[i]=\{\begin{array}{ll}1& ifK\le i\le N\\ 0& ifN<i\le K+W-1\end{array}$$

我们的目的是求出$dp[0]$，又因为每次拿牌是等可能的，因此有：
$$dp[i]=\frac{dp[i+1]+dp[i+2]+\cdots +dp[i+W]}{W}$$
合并起来的状态转移方程就是：
$$dp[i]=\{\begin{array}{ll}1& ifK\le i\le N\\ 0& ifN<i\le K+W-1\\ dp[i]=\frac{dp[i+1]+dp[i+2]+\cdots +dp[i+W]}{W}& if0\le i<K\end{array}$$

class Solution:
    def new21Game(self, N: int, K: int, W: int) -> float:
        dp=[0]*(K+W)
        s=0
        for i in range(K+W-1, -1, -1):
            if i >= K:
                dp[i] = 0 if i > N else 1
                s += dp[i]
            else:
                dp[i] = 1/W*s
                s -= dp[i+W]
                s += dp[i]
        return dp[0]