[KMP] Codeforces #631D. Messenger

本文介绍了一种用于处理字符串匹配问题的高效算法,并通过具体的代码实现展示了如何解决特定问题。该算法首先将输入字符串进行预处理,减少重复字符的影响,接着采用KMP算法的核心思想进行模式匹配,特别关注目标字符串两端字符的匹配情况。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

首先要把相同的缩到一起,然后考虑就是一个简单的匹配。
注意到b串两端的字符在a串里匹配可能比a串里那一块少。解决办法就是先把b的两端扣掉匹配,每次匹配成功时判断一下就好了。
b的长度2 直接特判掉。

#include<cstdio>
#include<algorithm>
#define Fir first
#define Sec second
#define mp(x,y) make_pair(x,y)
using namespace std;
const int maxn=200005;
typedef long long LL;
int n,m;
LL ans;
pair<LL,char> a[maxn],b[maxn];
void Pre(pair<LL,char> a[],int &n){
    int i=0; int _n=0;
    while(i<=n-1){
        int j=i+1; LL cnt=a[i].Fir;
        while(j<=n&&a[i].Sec==a[j].Sec) cnt+=a[j++].Fir;
        a[_n++]=mp(cnt,a[i].Sec);
        i=j;
    }
    n=_n;
}
int fail[maxn];
void make_kmp(){
    for(int i=2,j=0;i<=m-2;i++){
        while(j&&b[j+1]!=b[i]) j=fail[j]; 
        if(b[j+1]==b[i]) j++;
        fail[i]=j;
    }
}
void kmp(){
    for(int i=0,j=0;i<=n-1;i++){
        while(j&&b[j+1]!=a[i]) j=fail[j];
        if(b[j+1]==a[i]) j++;
        if(j==m-2){
            if(a[i+1].Sec==b[m-1].Sec&&a[i+1].Fir>=b[m-1].Fir&&a[i-m+2].Sec==b[0].Sec&&a[i-m+2].Fir>=b[0].Fir) ans++;
            j=fail[j];
        }
    }
}
int main(){
    freopen("cf631D.in","r",stdin);
    freopen("cf631D.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=0;i<=n-1;i++) scanf("%d-%c",&a[i].Fir,&a[i].Sec);
    for(int i=0;i<=m-1;i++) scanf("%d-%c",&b[i].Fir,&b[i].Sec);
    Pre(a,n); Pre(b,m);
    if(n<m) return printf("0"),0;
    if(m==1){
        ans=0;
        for(int i=0;i<=n-1;i++) if(a[i].Sec==b[0].Sec&&a[i].Fir>=b[0].Fir) ans+=a[i].Fir-b[0].Fir+1;
        printf("%I64d\n",ans);
        return 0;
    }
    if(m==2){
        ans=0;
        for(int i=0;i<=n-2;i++) 
         if(a[i].Sec==b[0].Sec&&a[i].Fir>=b[0].Fir&&a[i+1].Sec==b[1].Sec&&a[i+1].Fir>=b[1].Fir) ans++;
        printf("%I64d\n",ans);
        return 0;
    }
    make_kmp();
    kmp();
    printf("%I64d\n",ans);
    return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值