LeetCode 题解工作台

DI 序列的有效排列

给定一个长度为 n 的字符串 s ,其中 s[i] 是: “D” 意味着减少,或者 “I” 意味着增加 有效排列 是对有 n + 1 个在 [0, n] 范围内的整数的一个排列 perm ,使得对所有的 i : 如果 s[i] == 'D' ,那么 perm[i] > perm[i+1] ,以及; …

category

3

题型

code_blocks

5

代码语言

hub

3

相关题

当前训练重点

困难 · 状态·转移·动态规划

bolt

答案摘要

我们定义 表示字符串的前 个字符中,以数字 结尾的满足题目要求的排列的数量。初始时 ,其余 。答案为 $\sum_{j=0}^n f[n][j]$。 考虑 ,其中 $j \in [0, i]$。

Interview AiBox logo

Interview AiBox 实时 AI 助手,陪你讲清 状态·转移·动态规划 题型思路

试试 AiBox 面试助手arrow_forward
description

题目描述

给定一个长度为 n 的字符串 s ,其中 s[i] 是:

  • “D” 意味着减少,或者
  • “I” 意味着增加

有效排列 是对有 n + 1 个在 [0, n]  范围内的整数的一个排列 perm ,使得对所有的 i

  • 如果 s[i] == 'D',那么 perm[i] > perm[i+1],以及;
  • 如果 s[i] == 'I',那么 perm[i] < perm[i+1]

返回 有效排列  perm的数量 。因为答案可能很大,所以请返回你的答案对 109 + 7 取余

 

示例 1:

输入:s = "DID"
输出:5
解释:
(0, 1, 2, 3) 的五个有效排列是:
(1, 0, 3, 2)
(2, 0, 3, 1)
(2, 1, 3, 0)
(3, 0, 2, 1)
(3, 1, 2, 0)

示例 2:

输入: s = "D"
输出: 1

 

提示:

  • n == s.length
  • 1 <= n <= 200
  • s[i] 不是 'I' 就是 'D'
lightbulb

解题思路

方法一:动态规划

我们定义 f[i][j]f[i][j] 表示字符串的前 ii 个字符中,以数字 jj 结尾的满足题目要求的排列的数量。初始时 f[0][0]=1f[0][0]=1,其余 f[0][j]=0f[0][j]=0。答案为 j=0nf[n][j]\sum_{j=0}^n f[n][j]

考虑 f[i][j]f[i][j],其中 j[0,i]j \in [0, i]

如果第 ii 个字符 s[i1]s[i-1]'D',那么 f[i][j]f[i][j] 可以从 f[i1][k]f[i-1][k] 转移而来,其中 k[j+1,i]k \in [j+1, i],而由于 k1k-1 最大只能为 i1i-1,我们将 kk 向左移动一位,那么 k[j,i1]k \in [j, i-1],因此有 f[i][j]=k=ji1f[i1][k]f[i][j] = \sum_{k=j}^{i-1} f[i-1][k]

如果第 ii 个字符 s[i1]s[i-1]'I',那么 f[i][j]f[i][j] 可以从 f[i1][k]f[i-1][k] 转移而来,其中 k[0,j1]k \in [0, j-1],因此有 f[i][j]=k=0j1f[i1][k]f[i][j] = \sum_{k=0}^{j-1} f[i-1][k]

最终的答案即为 j=0nf[n][j]\sum_{j=0}^n f[n][j]

时间复杂度 O(n3)O(n^3),空间复杂度 O(n2)O(n^2)。其中 nn 是字符串的长度。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution:
    def numPermsDISequence(self, s: str) -> int:
        mod = 10**9 + 7
        n = len(s)
        f = [[0] * (n + 1) for _ in range(n + 1)]
        f[0][0] = 1
        for i, c in enumerate(s, 1):
            if c == "D":
                for j in range(i + 1):
                    for k in range(j, i):
                        f[i][j] = (f[i][j] + f[i - 1][k]) % mod
            else:
                for j in range(i + 1):
                    for k in range(j):
                        f[i][j] = (f[i][j] + f[i - 1][k]) % mod
        return sum(f[n][j] for j in range(n + 1)) % mod

我们可以用前缀和优化时间复杂度,使得时间复杂度降低到 O(n2)O(n^2)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution:
    def numPermsDISequence(self, s: str) -> int:
        mod = 10**9 + 7
        n = len(s)
        f = [[0] * (n + 1) for _ in range(n + 1)]
        f[0][0] = 1
        for i, c in enumerate(s, 1):
            pre = 0
            if c == "D":
                for j in range(i, -1, -1):
                    pre = (pre + f[i - 1][j]) % mod
                    f[i][j] = pre
            else:
                for j in range(i + 1):
                    f[i][j] = pre
                    pre = (pre + f[i - 1][j]) % mod
        return sum(f[n][j] for j in range(n + 1)) % mod

另外,我们也可以用滚动数组优化空间复杂度,使得空间复杂度降低到 O(n)O(n)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution:
    def numPermsDISequence(self, s: str) -> int:
        mod = 10**9 + 7
        n = len(s)
        f = [1] + [0] * n
        for i, c in enumerate(s, 1):
            pre = 0
            g = [0] * (n + 1)
            if c == "D":
                for j in range(i, -1, -1):
                    pre = (pre + f[j]) % mod
                    g[j] = pre
            else:
                for j in range(i + 1):
                    g[j] = pre
                    pre = (pre + f[j]) % mod
            f = g
        return sum(f) % mod
speed

复杂度分析

指标
时间Depends on the final approach
空间Depends on the final approach
psychology

面试官常问的追问

外企场景
  • question_mark

    Candidate understands dynamic programming concepts and is able to apply them to a string-based problem.

  • question_mark

    The candidate can model state transitions based on string patterns like 'I' and 'D'.

  • question_mark

    The candidate is capable of handling large numbers with modulo arithmetic.

warning

常见陷阱

外企场景
  • error

    Failing to properly handle the modulo operation, leading to overflow errors.

  • error

    Not accounting for all valid transitions when switching between 'I' and 'D' sequences.

  • error

    Overcomplicating the solution without recognizing that dynamic programming and state transitions can simplify the problem.

swap_horiz

进阶变体

外企场景
  • arrow_right_alt

    Modify the problem to work with permutations of a different set of integers, such as from 0 to n-1.

  • arrow_right_alt

    Introduce additional constraints, like allowing both 'I' and 'D' at the same index in the sequence.

  • arrow_right_alt

    Consider a similar problem where the goal is to find the number of valid permutations that satisfy multiple sequences of 'I' and 'D'.

help

常见问题

外企场景

DI 序列的有效排列题解:状态·转移·动态规划 | LeetCode #903 困难