题目链接: 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;
}