CodeForce #570 H. Subsequences

题目链接

 看题解学会了 $\theta (26n^2)$ 求不同子序列数量的 $dp$ 做法

 记 $lst[i][j]$ 为字母 ${(}^{\prime }{a}^{\prime }+j)$ 在下标 $i$ 前面最近出现的位置，不包括 $s[i]$ 本身。

 $dp[i][j]$ 为以 $s[i]$ 结尾，长度为 $j$ 的子序列数量。

 则 $dp[i][j]={\sum }_{k=0}^{25}dp[pos][j-1]$，其中 $pos=lst[i][k]$

 个人对转移方程的理解是：代表把前面以字符 ${(}^{\prime }{a}^{\prime }+k)$ 结尾的长度为 $j-1$ 的子序列接上 $s[i]$，因为结尾的字符不同所以不会有重复的。

剩下的就是贪心的从大到小枚举长度 $len$ ，计算长度为 $len$ 的子序列有多少，添加贡献，最后看所有序列（包括空串）是否有 $k$ 个。

int lst[110][26];
long long dp[110][110];
char s[110];
 
int main() {
    int n;
    long long K;
    scanf("%d %lld",&n,&K);
    scanf("%s",s);
 
    memset(lst,-1,sizeof(lst));
 
    // lst[0][ s[0] - 'a' ] = 0;
    for(int i=1;i<=n;i++) {
        for(int j=0;j<26;j++) {
            lst[i][j] = lst[i-1][j]; // ( s[i]-'a' == j ? i : lst[i-1][j] );
        }
        lst[i][s[i-1] - 'a'] = i-1;
    }
 
    for(int i=0;i<n;i++){
        dp[i][1] = 1;
        for(int j=2;j<=n;j++){
            for(int k=0;k<26;k++){
                int pos = lst[i][k];
                if( pos >= 0 ){
                    dp[i][j] += dp[pos][j-1];
                }
            }
        }
    }
 
    long long ans = 0, num = K;
    for(int i=n; i>0 && num > 0; i--) {
        long long cnt = 0;
        for(int j=0;j<26;j++) {
            int pos = lst[n][j];
            if( pos >= 0 ) {
                cnt = min( num, cnt + dp[pos][i] );
            }
        }
        ans += (n - i) * cnt;
        num -= cnt;
    }
 
    if( num > 0 ){
        num--;
        ans += n;
    }
 
    printf("%lld\n",num == 0 ? ans : -1);
}