C++ 中 DI 序列的有效排列


假设我们有一个字符串 S。这是一个来自集合 {'D', 'I'} 的字符字符串。(D 表示“递减”而 I 表示“递增”)

现在考虑一个有效排列是整数 {0 到 n} 的排列 P[0]、P[1]、…、P[n],使得对于所有 i,它满足以下规则

  • 如果 S[i] == 'D',则 P[i] > P[i+1];

  • 否则当 S[i] == 'I' 时,则 P[i] < P[i+1]。

我们必须找到有多少个有效的排列?答案可能非常大,因此我们将使用模 10^9 + 7 返回。

因此,如果输入类似于“IDD”,则输出将为 3,因此将有三个不同的排列,它们类似于 (0,3,2,1)、(1,3,2,0)、(2,3,1,0)。

为了解决这个问题,我们将遵循以下步骤 -

  • n := S 的大小

  • 定义一个大小为 (n + 1) x (n + 1) 的二维数组 dp

  • 用于初始化 j := 0,当 j <= n 时,更新(增加 j 1),执行 -

    • dp[0, j] := 1

  • 用于初始化 i := 0,当 i < n 时,更新(增加 i 1),执行 -

    • 如果 S[i] 与 'I' 相同,则 -

      • 用于初始化 j := 0,curr := 0,当 j < n - i 时,更新(增加 j 1),执行 -

        • curr := (dp[i, j] + curr) mod m

        • dp[i + 1, j] = (dp[i + 1, j] + curr)

    • 否则

      • 用于初始化 j := n - i - 1,curr := 0,当 j >= 0 时,更新(减少 j 1),执行 -

        • curr := (dp[i, j + 1] + curr) mod m

        • dp[i + 1, j] = (dp[i + 1, j] + curr)

  • 返回 dp[n, 0]

让我们看看以下实现以获得更好的理解 -

示例

 现场演示

#include <bits/stdc++.h>
using namespace std;
const int m = 1e9 + 7;
class Solution {
   public:
   int numPermsDISequence(string S) {
      int n = S.size();
      vector<vector<int>> dp(n + 1, vector<int>(n + 1));
      for (int j = 0; j <= n; j++)
      dp[0][j] = 1;
      for (int i = 0; i < n; i++) {
         if (S[i] == 'I') {
            for (int j = 0, curr = 0; j < n - i; j++) {
               curr = (dp[i][j] + curr) % m;
               dp[i + 1][j] = (dp[i + 1][j] + curr) % m;
            }
         } else {
            for (int j = n - i - 1, curr = 0; j >= 0; j--) {
               curr = (dp[i][j + 1] + curr) % m;
               dp[i + 1][j] = (dp[i + 1][j] + curr) % m;
            }
         }
      }
      return dp[n][0];
   }
};
main(){
   Solution ob;
   cout << (ob.numPermsDISequence("IDD"));
}

输入

"IDD"

输出

3

更新于: 2020年6月4日

677 次查看

开启您的 职业生涯

通过完成课程获得认证

开始
广告