牛客练习赛60 C-操作集锦 【动态规划】

传送门：C-操作集锦

---

题目描述
有一款自走棋有26种操作,每种操作我们都用 a,b,c,d,…,x,y,z 的符号来代替.
现在牛牛有一个长度为n的操作序列,他现在可以从里面拿出某些操作来组合成一个操作视频, 比如说操作序列是abcd,那么操作视频就有a,b,c,d,ab,ac,ad 等(也就是操作序列的子序列).他现在想知道长度为k且本质不同的操作视频有多少种.
比如对于abab，长度为 2 且本质不同的结果有 ab,aa,ba,bb。
考虑到答案可能非常大,你只需要输出在模 1e9+7 意义下的答案就可以了.

输入描述:
第一行两个整数 n, k.
第二行一个长度为n的字符串,保证只存在小写字母.

输出描述:
一行一个整数表示长度为 k 且本质不同的操作视频的个数.

示例1
输入

3 1
abc

输出

3

备注:

1≤n≤1e3
0≤k≤n

---

解题思路：

记 $dp[i][j]$ 表示前 i 个字符中长度为 j 的不同子序列个数。假设如果要求的不是不同序列，允许重复的话，应该有转移方程：
$dp[i][j]=dp[i-1][j]+dp[i-1][j-1]$

但是题目要求允许不同，那么上面的方程存在重复的子序列，问题出现在
$dp[i-1][j-1]$，因为 $dp[i-1][j]$ 可能出现的序列的结尾和 $s[i]$ 相同。所以我们只要减去这个重复的部分就行了。我们找到与当前结尾 $s[i]$ 相同的之前一个字符的位置，我们记录 $pre[s[i]]$ 表示 $s[i]$ 这个字符之前出现的位置,那么多算的这部分很容易发现就是 $dp[pre[s[i]]-1][j-1]$。

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<cmath>
#include<cstring>
#define ll long long 
using namespace std;
const int N = 1e3+10;
const int mod = 1e9 + 7;
 
ll dp[N][N];
 
int n, k, pre[N];
char s[N];
 
int main()
{
    scanf("%d%d", &n, &k);
    scanf("%s", s + 1);
    dp[0][0] = 1;
    for (int i = 1; i <= n; i++) 
    {
        dp[i][0] = 1;
        for (int j = 1; j <= i; j++) 
        {
            dp[i][j]=dp[i - 1][j]+dp[i - 1][j - 1];
            if (pre[s[i] - 'a'])
                dp[i][j] -= dp[pre[s[i] - 'a'] - 1][j - 1];
            dp[i][j]%= mod;
        }
        pre[s[i] - 'a'] = i;
    }
    ll res = dp[n][k];
    if (res < 0)
        res += mod;
    printf("%lld\n", res);
    return 0;
}