CF-1183H Subsequences (hard version)(DP)

题目链接： https://codeforc.es/problemset/problem/1183/H

题目大意：

给你一个字符串，删去一个字符需要花费1个价值，删去字符后就可以得到新的子串，要使得子串集合的大小为k，最小的代价是多少。

我们定义dp[i][len]为以第i个字母为子串最后一个字符，长度为len的子串有多少个。

每一个字母由上一个字母转移得到。

如：如果第i个字符为‘y’,那么 dp[i][len] += dp[前i-1个字符中最后的'a'-'z'][len-1];

递归边界是dp[i][1] = 1;

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<unordered_map>
using namespace std;

int last[105][30];
long long dp[105][105];
char str[105];

int main()
{
    int n;
    long long k;
    scanf("%d%lld", &n, &k);
    scanf("%s", str+1);

    memset(last,-1,sizeof(last));
    for(int i = 1; i <= n; i++)
    {
        for(int j = 0; j < 26; j++)
        {
            last[i][j] = last[i-1][j];
        }
        last[i][(int)(str[i]-'a')] = i;
    }

    memset(dp,0,sizeof(dp));
    for(int i = 1; i <= n; i++)
        dp[i][1] = 1;

    for(int len = 2; len < n; len++)
    {
        for(int i = 2; i <= n; i++)
        {
            for(int j = 0; j < 26; j++)
            {
                if(last[i-1][j]!=-1)
                    dp[i][len] = dp[i][len]+dp[last[i-1][j]][len-1];
            }
        }
    }

    long long ans = 0;
    k--;  //有一个长度为n的串是不需要代价的
    for(int len = n-1; len>0; len--)
    {
//        cout <<len<<" "<<k<<endl;
        long long cnt = 0;
        for(int i = 0; i < 26; i++)
        {
            if(last[n][i]!=-1)
                cnt+=dp[last[n][i]][len];
        }
        if(cnt>k)
        {
            ans+=(n-len)*k;
            k = 0;
            break;
        }
        else
        {
            ans+=(n-len)*cnt;
            k-=cnt;
        }
    }
    if(k==1)
    {
        ans+=n;
        k=0;
    }

    if(k==0)
        printf("%lld\n", ans);
    else
        printf("-1\n");



    return 0;
}