Consecutive Wins

https://binarysearch.com/problems/Consecutive-Wins

题目描述

You are given integers n and k. Given that n represents the number of games you will play, return the number of ways in which you win k or less games consecutively. Mod the result by 10 ** 9 + 7.

Constraints

n ≤ 10,000
k ≤ 10
Example 1
Input
n = 3
k = 2
Output
7
Explanation
Here are the ways in which we can win 2 or fewer times consecutively:

"LLL"
"WLL"
"LWL"
"LLW"
"WWL"
"LWW"

前置知识

  • 递归树

  • 动态规划

公司

  • 暂无

思路

定义 f(i, j) 表示 i 次比赛连续赢 j 次的总的可能数目。 其实也就是长度为 n - i 的字符串中连续 w 的次数不超过 j 的总的可能数,其中 字符串中的字符只能是 L 或 W。

经过这样的定义之后,我们的答案应该是 f(0, 0)。

实际上,也就是问动态规划的转移方程是什么。

对于 f(0, 0),我们可以:

  • 在末尾增加一个 L,也就是输一局。用公式表示就是 f(1, 0)

  • 在末尾增加一个 W,也就是赢一局。用公式表示就是 f(1, 1)

用图来表示就是如下的样子:

不是一般性,我们可以得出如下的转移方程:

f(i,j)={f(i+1,0)+f(i+1,j+1)j<kf(i+1,0)j=kf(i, j)=\left\{ \begin{aligned} f(i+1, 0) + f(i+1, j+1) & & j < k \\ f(i+1, 0) & & j = k \\ \end{aligned} \right.

那么我们的目标其实就是计算图中叶子节点(绿色节点)的总个数。

注意 f(3,3) 是不合法的,我们不应该将其计算进去。

上面使我们的递归树代码,可以看出有很多重复的计算。这提示我们使用记忆化递归来解决。使用记忆化递归,总的时间复杂度 节点总数 * 单个节点的操作数。树的节点总数是 n * k,单个节点的操作是常数。故总的时间复杂度为 $O(n * k),空间复杂度是使用的递归深度 + 记忆化使用的额外空间。其中递归深度是 $n$,记忆化的空间为 $n * k$,忽略低次项后空间复杂度为 $O(n * k)$

代码

代码支持:Python3

Python3 Code:

class Solution:
    def solve(self, n, k):
        @lru_cache(None)
        def dp(i, cnt):
            if i == n:
                return 1
            ans = dp(i + 1, 0)  # place L
            if cnt < k:
                ans += dp(i + 1, cnt + 1)  # place W if I can
            return ans

        return dp(0, 0) % (10 ** 9 + 7)

复杂度分析

  • 时间复杂度:$O(n * k)$

  • 空间复杂度:$O(n * k)$

最后更新于