[LeetCode]837. 新21点

题目

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

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

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

示例 1：

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

示例 2：

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

示例 3：

输入：N = 21, K = 17, W = 10
输出：0.73278

提示：

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

解题思路

简单分析一下题目，得分少于 K 分时抽取数字，得分大于等于 K 分时停止抽取数字。 抽取数字时，从 [1, W] 的范围中随机获得一个整数作为分数进行累计（可以重复抽），求最终分数小于等于 N 的概率是多少？
设 dp[x] 为现有牌面分数为x时，能获胜的概率。
x 最多能到 K-1，因为当大于等于 K 时，爱丽丝会停止抽牌，所以当游戏结束时，即爱丽丝停止抽牌时，她可能达到的最大牌面是 K-1+W，而一开始她的牌面是 0，所以我们用一个长度为 K+W 的 dp 数组来保存她在所有面值下的胜率。最后 dp[0]，也就是最开始爱丽丝还没有抽牌，她的牌面为 0 时的胜率，这个就是我们要求的答案。
所以先将数组分成两部分 [0,K-1] 和 [K,K+W-1]，区别就是 [0,K-1] 时爱丽丝可以抽牌， [K,K-1+W] 时不能抽牌，那么不能抽牌时她获胜的概率是多少呢，很显然牌面小于等于 N 时，概率就是 1，大于 N 概率就是 0，所以可以先直接填满[ K,K-1+W]。
接下来，从 K-1 开始填 [0,K-1] 部分，因为小于 K 时可以抽牌，且牌面机会都是均等的，她能抽取的面值在 [1,W] 之间，所以将概率之和平均一下就是 dp[x] 的概率：

dp[x] = 1/w * (dp[x+1]+dp[x+2]+dp[x+3]...+dp[x+w])

当计算下一轮的 dp[x-1] 时，我们可以用 O(1) 的时间复杂度计算 sum(dp[x]……dp[x+W-1])
方法就是先减去最右边的 dp[x+w] ，再加上一个左边的 dp(x) 即可。
复杂度分析：
时间复杂度：O(K+W)，即dp数组长度。
空间复杂度：O(K+W)，即dp数组长度。

代码

class Solution {
    public double new21Game(int N, int K, int W) {
        double[] dp = new double[K + W];
        double S = 0;
        for(int i=K; i<K+W; i++){
            if(i<=N){
                dp[i] = 1;
                S += dp[i];
            }
            // else{
            //     dp[i] = 0; // 这句写或不写都可以，因为java数组初始化时就是0。
            // }
        }
        for(int i=K-1; i>=0; i--){
            dp[i] = S/W;
            S = S - dp[i+W] + dp[i];
        }
        return dp[0];
    }
}