Cod sursă (job #742865)

#include <bits/stdc++.h>

using namespace std;

ifstream fin("director.in");
ofstream fout("director.out");

const int mod=(int)1e9+7;

int n,k;

string s;

int last[26];
int sum;
int dp[1005][5005];

int main()
{
    fin>>n>>k>>s;
    s=" "+s;

    for(int i=1; i<=n; i++)
        dp[1][i]=1;
    for(int j=2; j<=k; j++)
    {
        fill(last,last+26,0);

        sum=0;
        for(int i=1; i<=n; i++)
        {
            dp[j][i]=sum;
            sum=(sum-dp[j-1][last[s[i]-'a']]+mod)%mod;
            last[s[i]-'a']=i;
            sum=(sum+dp[j-1][last[s[i]-'a']])%mod;
        }
    }

    int ans=0;
    for(int i=0; i<26; i++)
        ans=(ans+dp[k][last[i]])%mod;
    fout<<ans<<"\n";
    return 0;
}