Leetcode837.新21点

目录

      • 题目
      • [算法标签: 数学, 概率, 动态规划](#算法标签: 数学, 概率, 动态规划)
      • 思路
      • 代码

题目

837. 新 21 点

算法标签: 数学, 概率, 动态规划

思路

定义状态表示为 f [ i ] f[i] f[i], 表示分数达到 i i i的时候的概率, 分析状态计算, 假设当前的分数是 i i i, 抽取到的牌得分数是 x x x, 那么当前状态就会转移到 f [ i + x ] f[i + x] f[i+x], 状态转移方程如下

d p [ i ] = 1 maxPts ( d p [ i + 1 ] + d p [ i + 2 ] + ⋯ + d p [ i + maxPts ] ) dp[i] = \frac{1}{\text{maxPts}} \left( dp[i+1] + dp[i+2] + \cdots + dp[i+\text{maxPts}] \right) dp[i]=maxPts1(dp[i+1]+dp[i+2]+⋯+dp[i+maxPts])

计算时间复杂度, 外层枚举分数, 内层也需要枚举分数, 总的时间复杂度来到了 O ( n 2 ) O(n ^ 2) O(n2), 时间复杂度过高, 需要进行优化, 推 i = i − 1 i = i - 1 i=i−1时的表达式

d p [ i − 1 ] = 1 maxPts ( d p [ i ] + d p [ i + 1 ] + ⋯ + d p [ i + maxPts - 1 ] ) dp[i - 1] = \frac{1}{\text{maxPts}} \left( dp[i] + dp[i+1] + \cdots + dp[i+\text{maxPts - 1}] \right) dp[i−1]=maxPts1(dp[i]+dp[i+1]+⋯+dp[i+maxPts - 1])

设 t = m a x P t s t = maxPts t=maxPts, f [ i ] = f [ i + 1 ] × t + f [ i + 1 ] − f [ i + t + 1 ] t f[i] = \frac {f[i + 1] \times t + f[i + 1] - f[i + t + 1]}{t} f[i]=tf[i+1]×t+f[i+1]−f[i+t+1], 整理后得到

f [ i ] = f [ i + 1 ] + f [ i + 1 ] − f [ i + t + 1 ] t f[i] = f[i + 1] + \frac {f[i + 1] - f[i + t + 1]} {t} f[i]=f[i+1]+tf[i+1]−f[i+t+1]

这样就将时间复杂度降低到 O ( n ) O(n) O(n)

代码

cpp 复制代码
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>

using namespace std;

const int N = 2e4 + 10;

class Solution {
public:
    double new21Game(int n, int k, int maxPts) {
        if (k == 0) return 1.0;

        //当前分数是i, 并且分数不超过n的概率
        double f[N] = {0};
        for (int i = k; i <= n && i < k + maxPts; ++i) f[i] = 1.0;

        //计算当前分数是i再抽一张牌, 得分不超过n的概率
        f[k - 1] = 1.0 * min(n - k + 1, maxPts) / maxPts;

        for (int i = k - 2; i >= 0; --i) {
            f[i] = f[i + 1] + (f[i + 1] - f[i + maxPts + 1]) / maxPts;
        }

        return f[0];
    }
};
相关推荐
不知天地为何吴女士1 小时前
Day32| 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯
算法
小坏坏的大世界1 小时前
C++ STL常用容器总结(vector, deque, list, map, set)
c++·算法
liulilittle2 小时前
C++ TAP(基于任务的异步编程模式)
服务器·开发语言·网络·c++·分布式·任务·tap
励志要当大牛的小白菜4 小时前
ART配对软件使用
开发语言·c++·qt·算法
qq_513970444 小时前
力扣 hot100 Day56
算法·leetcode
PAK向日葵5 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男7 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
_Kayo_7 小时前
VUE2 学习笔记6 vue数据监测原理
vue.js·笔记·学习
YuTaoShao7 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
YouQian7727 小时前
Traffic Lights set的使用
算法