【牛客练习赛60：C】操作集锦（dp+子序列计数）

传送门

• 题目
  给出长度为的字符串，求有多少种不同的长度为的子序列。$(1\le n\le 1e3,0\le k\le n)$
• 思路

  • 空串也是一种合法的子序列，所以特判$k=0$

  • 二维dp求解(当然也可以一维，我原本的做法是一维的，比下面的解法稍微麻烦一点，就不讲啦)

    • 解法一：$O(26n^2)$
      $dp[i][j]$表示前$i$个字符中长度为$j$且以$s[i]$为结尾的子序列种类数$(1\le i\le n,1\le j\le i)$
      $nxt[i][j]$表示$i$后面$j{+}^{\prime }{a}^{\prime }$第一次出现的位置$(0\le i<n,0\le j<26)$，这样做是为了避免重复统计答案。
      转移方程：

    $dp[x][j+1]+=dp[i][j]$ $(x=nxt[i][p]\&\&x!=0,0\le p<26)$
    $ans={\sum }_{i=k}^{n}dp[i][k]$
    注意初始化：$dp[0][0]=1$

    • 解法二：$O(n^2)$
      $dp[i][j]$表示前$i$个字符中长度为$j$的子序列种类数$(1\le i\le n,1\le j\le i)$
      $last[i]$表示$i{+}^{\prime }{a}^{\prime }$上一次出现的位置$(0\le i<26)$
      转移方程：
      $dp[i][j]=\{\begin{array}{ll}dp[i-1][j]+dp[i-1][j-1]& last[s[i]{-}^{\prime }{a}^{\prime }]==0\\ dp[i-1][j]+dp[i-1][j-1]-dp[x-1][j-1]& x=last[s[i]{-}^{\prime }{a}^{\prime }]!=0\end{array}$
      注意为了防止dp[i][j]的结果为负数，需要+mod再取模

    $ans=dp[n][k]$

    注意初始化$dp[i][0]=1$ $(0\le i\le n)$

• ac代码：
  • 解法一：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
const int maxn = 1e3+10;
ll dp[maxn][maxn];
int nxt[maxn][30];
int n, k;
char s[maxn];
int main()
{
    scanf("%d %d", &n, &k);
    scanf("%s", s+1);
    if(k==0) {printf("1\n"); return 0;}
    for(int i = 0; i < n; i++)
        for(int j = i+1; j <= n; j++)
            if(!nxt[i][s[j]-'a']) nxt[i][s[j]-'a'] = j;
    dp[0][0] = 1;
    for(int i = 0; i <= n; i++)
    {
        for(int j = 0; j <= min(i, k); j++)
        {
            for(int p = 0; p < 26; p++)
            {
                int x = nxt[i][p];
                if(x!=0) dp[x][j+1] = (dp[x][j+1]+dp[i][j])%mod;
            }
        }
    }
    ll ans = 0;
    for(int i = k; i <= n; i++) ans = (ans+dp[i][k])%mod;
    printf("%lld\n", ans);
    return 0;
}

• 解法二：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
const int maxn = 1e3+10;
ll dp[maxn][maxn];
int last[maxn];
int n, k;
char s[maxn];
int main()
{
    scanf("%d %d", &n, &k);
    scanf("%s", s+1);
    if(k==0) {printf("1\n"); return 0;}
    dp[0][0] = 1;
    for(int i = 1; i <= n; i++)
    {
        dp[i][0] = 1;
        for(int j = 1; j <= min(i, k); j++)
        {
            dp[i][j] = (dp[i-1][j]+dp[i-1][j-1])%mod;
            if(last[s[i]-'a']!=0) dp[i][j] = (dp[i][j]-dp[last[s[i]-'a']-1][j-1]+mod)%mod;
        }
        last[s[i]-'a'] = i;
    }
    printf("%lld\n", dp[n][k]%mod);
    return 0;
}