HDU 6068 Classic Quotation

答案分成三份,L左边完整的串,R右边完整的串,以及中间拼凑的串。
S[0][L][i]表示T串的前缀0~L中,有哪些位置能够匹配到P[i]。
而S[1][R][j]表示T串翻转后,后缀R~n-1变为前缀0~n-R-1中,有哪些位置能够匹配到P串翻转后的P[j].
可以在KMP中通过失配函数维护S数组,S[][i][j] = s[][i-1][j]+(T[i]能否刚好匹配到p[j])?1:0, 复杂度O(n*m);
这样只要枚举一遍P串,求得sum(S[0][L][i]*S[1][R][m-i])为拼凑的串的总和。
剩余的串即sum(S[0][i][m] | 0<=i<=L)+sum(S[1][i][m] | 0<=i<=n-R-1).

代码:

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

#define pb push_back
#define fi first
#define se second
#define CLR(A, X) memset(A, X, sizeof(A))
typedef long long LL;
typedef pair<int, int> PII;
const double eps = 1e-10;
int dcmp(double x) { if(fabs(x) < eps) return 0; return x>0?1:-1; }
const LL MOD = 1e9+7;
const int N = 5e4+5, M = 1e2+5;
const int INF = 0x3f3f3f3f;

char T[N], P[M];
int f[2][M];
LL S[2][N][M];

void getFail(int m, int *f) {
    f[0] = 0, f[1] = 0;
    for(int i = 1; i < m; i++) {
        int j = f[i];
        while(j && P[i]!=P[j]) j = f[j];
        f[i+1] = P[i]==P[j]?j+1:0;
    }
}

void KMP(int n, int m, int *f, LL S[][M]) {
    getFail(m, f);
    int j = 0;
    for(int i = 0; i < n; i++) {
        for(int k = 0; k <= m; k++) {
            if(!i) S[i][k] = 0;
            else S[i][k] = S[i-1][k];
        }
        while(j && T[i]!=P[j]) j = f[j];
        if(T[i] == P[j]) j++;
        int x = j;
        while(x) {
            S[i][x]++;
            x = f[x];
        }
        if(j == m) j = f[j];
    }
    for(int i = 1; i < n; i++) S[i][m] += S[i-1][m];
}

int main() {
    int cas;
    scanf("%d", &cas);
    while(cas--) {
        int n, m, q, L, R;
        scanf("%d%d%d%s%s", &n, &m, &q, T, P);
        KMP(n, m, f[0], S[0]);
        reverse(T, T+n); reverse(P, P+m);
        KMP(n, m, f[1], S[1]);

        while(q--) {
            scanf("%d%d", &L, &R);
            L--, R--;
            LL ans = S[0][L][m]*(n-R)+S[1][n-R-1][m]*(L+1);
            for(int i = 1; i < m; i++) {
                ans += S[0][L][i]*S[1][n-R-1][m-i];
            }
            printf("%lld\n", ans);
        }
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值