【leetcode】903 DI序列的有效排列(动态规划)

题目链接:https://leetcode-cn.com/problems/valid-permutations-for-di-sequence/

题目描述

我们给出 S,一个源于 {'D', 'I'} 的长度为 n 的字符串 。(这些字母代表 “减少” 和 “增加”。)
有效排列 是对整数 {0, 1, ..., 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.

示例:

输入:"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)

提示:

  1. 1 <= S.length <= 200
  2. S 仅由集合 {‘D’, ‘I’} 中的字符组成。

思路

https://www.cnblogs.com/grandyang/p/11094525.html
以[0,1,2,3]组成’DID’模式为例:
dp[0][0]=1

加第二个数字需要降序,加的数字不能大于最后一个数字0,则只能加0
dp[1][0]=1 (0->1->10)

加第三个数字需要升序,加的数字大于最后一个数字0,实际可加数字有1,2
加1: ( dp[2][1] = 1 )
10 -> 20 -> 201
加2: ( dp[2][2] = 1 )
10 -> 10 -> 102

加第四个数字需要降序,加的数字不能大于最后一个数字,上一轮最后一个数字有1或2
则实际可加数字有0,1,2
加0:(dp[3][0] = 2)
201 -> 312 -> 3120
102 -> 213 -> 2130

加1: ( dp[3][1] = 2 )
201 -> 302 -> 3021
102 -> 203 -> 2031

加2: ( dp[3][2] = 1 )
102 -> 103 -> 1032

这种方法算出的 dp 数组为:
1 0 0 0
1 0 0 0
0 1 1 0
2 2 1 0
最后把 dp 数组的最后一行加起来 2+2+1 = 5 就是最终的结果

状态方程为
if (S[i-1] == ‘D’) dp[i][j] += dp[i-1][k] ( j <= k <= i-1 )
else dp[i][j] += dp[i-1][k] ( 0 <= k < j )

class Solution {
public:
    int numPermsDISequence(string S) {
        int res = 0, n = S.size(), M = 1e9 + 7;
        vector<vector<int>> dp(n + 1, vector<int>(n + 1));
        dp[0][0] = 1;
        for (int i = 1; i <= n; ++i) {
            for (int j = 0; j <= i; ++j) {
                if (S[i - 1] == 'D') {
                    for (int k = j; k <= i - 1; ++k) {
                        dp[i][j] = (dp[i][j] + dp[i - 1][k]) % M;
                    } 
                } else {
                    for (int k = 0; k <= j - 1; ++k) {
                        dp[i][j] = (dp[i][j] + dp[i - 1][k]) % M;
                    }
                }
            }
        }
        for (int i = 0; i <= n; ++i) {
            res = (res + dp[n][i]) % M;
        }
        return res;
    }
};

你可能感兴趣的:(LeetCode)