2021 MINIEYE杯 杭电多校5 1004 Another String 字符串匹配

1004 Another String

https://acm.hdu.edu.cn/showproblem.php?pid=7015

题意：定义两个长度相同的字符串之间的距离为这两个字符串中相同位置字符不同的位置数量。如果相同长度的两个字符串的距离不超过 k，称这两个字符串满足 k-匹配。给定一个长度为 n 的字符串 S 和一个整数 k。对于每个 i∈[1,n−1]，将 S 分成子串 A 和 B，而 A=S[1,i] 和 B=S[i+1,n]。对于由 A 的一些非空子串和 B 的一些非空子串形成的所有字符串对，计算满足 k-匹配的对数。

分析：定义f[i][j]为字符串S分别从i起始和从j起始的子串中满足k匹配的最大长度。可以外层循环枚举d=|j-i|，内层循环通过双指针法线性计算出f[i][j]，时间复杂度为O(n^2)，具体做法见代码。定义t分割是将S分为A[1,t]和B[t+1,n]，则要统计的答案为。发现t的答案可以传递给t'=t-1，所以考虑从大到小枚举t。设cnt[i][f[i][j]]为对于每个i，当j取遍t+1到n时，k匹配最长长度为f[i][j]，也即在当前t下左端点为i，j的k匹配数量。每次可以通过cnt数组来计算答案的变化，即删除长度越界的数量。

#include <bits/stdc++.h>
using namespace std;

const int N = 3005;

int n, k, f[N][N], cnt[N][N], res[N];
char s[N];

void solve() {
    scanf("%d%d%s", &n, &k, s);
    memset(cnt, 0, sizeof cnt);
    for (int d = 1; d < n; d++) {
        int dif = 0;
        for (int i = 0, j = i + d, len = 0; j < n; i++, j++) {
            while (j + len < n && dif <= k) {
                if (s[i + len] - s[j + len]) dif++;
                len++;
            }
            f[i][j] = len - (dif > k);
            dif -= s[i] != s[j];
            len--;
        }
    }
    int ans = 0;
    for (int t = n - 2; t >= 0; t--) {
        for (int i = 0; i <= t; i++) {
            ans -= cnt[i][t - i + 2];
            cnt[i][t - i + 1] += cnt[i][t - i + 2];
        }
        for (int i = 0; i <= t; i++) {
            int len = min(f[i][t + 1], t - i + 1);
            ans += len;
            cnt[i][len]++;
        }
        ans -= cnt[t + 1][1];
        res[t] = ans;
    }
    for (int i = 0; i < n - 1; i++) printf("%d\n", res[i]);
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) solve();
    return 0;
}