Cod sursă (job #742867)

#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[2][10005];

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

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

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

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