「CF535D」Tavas and Malekas「KMP」

题意

一个有小写字母组成的长度为 n n 文本串,给出匹配串及匹配的m个位置(按顺序给出),求原串有多少种可能.

题解

可以转化为判定这些匹配点有没有冲突。有冲突就是 0 0 ,没有冲突就是26k k k 是没有被匹配覆盖的位置个数.

显然两个匹配有冲突他们一定有重叠。重叠的一定是匹配串的后缀和前缀。所以假设两个匹配位置他们重叠的部分长度为x,我们只需要判断匹配串长度为 x x 的前缀和后缀是否相等即可.

先预处理出next数组。
可以使用 nextl n e x t l l l <script type="math/tex" id="MathJax-Element-10">l</script>为匹配串长度)一直跳,经过的所有位置都标记一下合法.
这样预处理完就可以直接判断了.

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int MAXN = 1000010;
const int MOD = 1000000007;

int n, m, l;
int pos[MAXN];
char s[MAXN];

int f[MAXN];
bool tag[MAXN];

void Get_fail() {
    f[0] = f[1] = 0;
    for(int i = 1; i < l; i ++) {
        int j = f[i];
        for(; j && s[i] != s[j]; j = f[j]) ;
        f[i + 1] = s[i] == s[j] ? j + 1 : 0;
    }
    tag[l] = true;
    for(int i = f[l]; i; i = f[i]) tag[i] = true;
}

int Qpow(int a, int b) {
    int ans = 1;
    for(; b; b >>= 1, a = a * 1ll * a % MOD)
        if(b & 1) ans = ans * 1ll * a % MOD;
    return ans;
}

int main() {
    scanf("%d%d%s", &n, &m, s);
    l = strlen(s);
    Get_fail();
    for(int i = 1; i <= m; ++ i) {
        scanf("%d", &pos[i]);
        -- pos[i];
    }
    for(int i = 1; i <= m; ++ i) {
        if(pos[i] + l > n) return puts("0"), 0;
        if(i > 1 && pos[i - 1] + l > pos[i]) {
            int x = pos[i - 1] + l - pos[i];
            if(!tag[x]) return puts("0"), 0;
        }
    }
    int len = 0, L = 0, R = -1;
    for(int i = 1; i <= m; ++ i) {
        if(i > 1 && pos[i - 1] + l > pos[i]) R = pos[i] + l - 1;
        else {
            len += R - L + 1;
            L = pos[i], R = pos[i] + l - 1;
        }
    }
    len += R - L + 1;
    printf("%d\n", Qpow(26, n - len));
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值