codeforces1183H Subsequences (hard version)(线性DP)

codeforces1183H Subsequences (hard version)(线性DP),第1张

codeforces1183H Subsequences (hard version)(线性DP)

题目链接:codeforces1183H Subsequences (hard version)

题目思路:

求前 k k k 长的子序列。定义dp[i][j] 为前 i i i 个长度为 j j j 的子序列个数,首先将 dp[i-1][j] 的状态转移到 dp[i][j],再加上不包含当前 s[i] 的子序列个数,即 dp[i-1][j-1]。又因为前面可能会有 s[last] 与 s[i] 相等,我们只要减去 dp[last-1][j-1] 即可,last 是在 i i i 前面离 i i i 最近的相等字符的下标。

参考代码:

#include 
using namespace std;
typedef long long ll;
char s[110];
ll dp[110][110];
int pre[110];
int main() {
    ll n, k;
    cin >> n >> k >> s+1;
    for (int i = 1; i <= n; i++) {
        for (int j = i+1; j <= n; j++) {
            if (s[i] == s[j]) pre[j] = i;
        }
    }
    for (int i = 0; i <= n; i++) {
        dp[i][0] = 1;
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= i; j++) {
            dp[i][j] = dp[i-1][j] + dp[i-1][j-1];
            if (pre[i] != 0) dp[i][j] -= dp[pre[i]-1][j-1];
        }
    }
    ll ans = 0;
    for (int i = n; i >= 1 && k; i--) {
        ans += min(dp[n][i], k) * (n-i);
        k = max(0LL, k-dp[n][i]);
    }
    if (k > 1) cout << -1 << endl;
    else {
        if (k == 1) ans += n;
        cout << ans << endl;
    }
    return 0;
}

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5611228.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-15
下一篇 2022-12-15

发表评论

登录后才能评论

评论列表(0条)

保存