python反序数如何编写_Python 动态规划，反序求解

题目1：新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

提示：

0 <= K <= N <= 10000

1 <= W <= 10000

如果答案与正确答案的误差不超过 10^-5，则该答案将被视为正确答案通过。

此问题的判断限制时间已经减少。

Leetcode:力扣​leetcode-cn.com

方法：动态规划，反序求解

1.设dp[i]表示从i开始抽取数字，分数不超过N的概率

2.初始化后(K,min(N+1,K+W))个值，大于N的概率为0，小于N的概率为1

3.状态方程：因为后面的状态对前面产生影响，所以反向求解

dp[i] = (dp[i+1]+...+dp[i+W])/W

时间复杂度为O(min(N,K+W))，空间复杂度为O(K+W)

class Solution:

def new21Game(self, N: int, K: int, W: int) -> float:

dp = [0.0]*(K+W)

for i in range(K,min(N+1,K+W)):

dp[i] = 1.0

S = min(N-K+1,W)

for i in range(K-1,-1,-1):

dp[i] = S/float(W)

S += dp[i]-dp[i+W]

return dp[0]

题目2：

在一个火车旅行很受欢迎的国度，你提前一年计划了一些火车旅行。在接下来的一年里，你要旅行的日子将以一个名为 days 的数组给出。每一项是一个从 1 到 365 的整数。

火车票有三种不同的销售方式：

一张为期一天的通行证售价为 costs[0] 美元；

一张为期七天的通行证售价为 costs[1] 美元；

一张为期三十天的通行证售价为 costs[2] 美元。

通行证允许数天无限制的旅行。 例如，如果我们在第 2 天获得一张为期 7 天的通行证，那么我们可以连着旅行 7 天：第 2 天、第 3 天、第 4 天、第 5 天、第 6 天、第 7 天和第 8 天。

返回你想要完成在给定的列表 days 中列出的每一天的旅行所需要的最低消费。

leetcode:力扣​leetcode-cn.com

方法1：动态规划，正向求解

设dp[i]表示第i天的最低消费

初始化dp[i]全为0

关系式:

if i == days[i]:

dp[i] = min(dp[max(0,i-1)]+costs[0],dp[max(0,i-7)]+costs[1],dp[max(0,i-30)]+costs[2])

else:

dp[i] = dp[i-1]

class Solution:

def mincostTickets(self, days: List[int], costs: List[int]) -> int:

dp = [0]*(days[-1]+1)

days_index = 0

for i in range(1, len(dp)):

if days[days_index]==i:

dp[i] = min(dp[max(0,i-1)]+costs[0],

dp[max(0,i-7)]+costs[1],

dp[max(0,i-30)]+costs[2])

days_index +=1

else:

dp[i] = dp[i-1]

return dp[-1]

方法2：动态规划，反向求解

设dp[i]表示从第i天到最后一天的最低消费

关系式：

如果i in days:

dp[i] = min(dp[i+1]+costs[0], dp[i+7]+costs[1], dp[i+30]+costs[2])

否则：

dp[i] = dp[i+1]

class Solution:

def mincostTickets(self, days: List[int], costs: List[int]) -> int:

dp = [0]*(days[-1]+31)

for i in range(days[-1],-1,-1):

if i in days:

dp[i] = min(dp[i+1]+costs[0], dp[i+7]+costs[1], dp[i+30]+costs[2])

else:

dp[i] = dp[i+1]

return dp[0]

ps:共同进步，欢迎大家批评指正!