CodeForces - 1183E（dp）

题目链接：https://vjudge.net/contest/349486#problem/C

 

思路：

如果字符串没有重复字符，dp(i,j) = dp(i-1,j-1) + dp(i,j-1)

dp(i,j)表示长度为i，以j字符结尾的字符串的个数。

考虑重复的情况，如果重复，减去之前i-1长度串中ss[j]的影响dp(i,j) -= dp(i-1,pre(ss[j]) );

 

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 220;
ll dp[N][N] = {0};
int mp[N];
char ss[N];
int main(void)
{
    int n,m;
    scanf("%d%d",&n,&m);
    scanf("%s",ss+1);
    for(int i=0;i<=n;i++) dp[0][i] = 1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=25;j++) mp[j] = 0;
        for(int j=i;j<=n;j++)
        {
            dp[i][j] = dp[i-1][j-1] + dp[i][j-1];
            if(mp[ ss[j] - 'a' ] != 0)
                dp[i][j] -= dp[i-1][ mp[ ss[j] - 'a' ]-1 ];
            mp[ ss[j] - 'a' ] = j;
        }
    }
    ll sum = 0,ans = 0;
    for(int i=n;i>=0;i--)
    {
        if(sum + dp[i][n] >= m)
        {
            ans += (m - sum)*(n-i);
            sum = m;
            break;
        }
        else{
            ans += dp[i][n]*(n-i);
            sum += dp[i][n];
        }
    }
    if(sum >= m)
        printf("%lld\n",ans);
    else printf("-1\n");
    return 0;
}