codeforces 631D Messenger - KMP算法

传送门
题解:模板题,分类讨论。

#include<iostream>
#include<cstring>
#include<cstdio>
#define N 1000010
#define lint long long
using namespace std;
int nxt[N],pos[N];lint cs[N],ct[N];
char s[N],t[N];
inline int get_nxt(char *s,lint *c,int n)
{
//  for(int i=1;i<=n;i++)
//      printf("%lld-%c ",c[i],s[i]);
//  printf("\n");
    nxt[1]=0;
    for(int i=2,j;i<=n;i++)
    {
        j=nxt[i-1];
        while(j&&(s[i]!=s[j+1]||c[i]!=c[j+1])) j=nxt[j];
        if(s[i]==s[j+1]&&c[i]==c[j+1]) j++;nxt[i]=j;
    }
    return 0;
}
inline int KMP(char *s,lint *c1,char *t,lint *c2,int n,int m)
{
    int c=0;
    for(int i=1,j=0;i<=n;i++)
    {
        while(j&&(s[i]!=t[j+1]||c1[i]!=c2[j+1])) j=nxt[j];
        if(s[i]==t[j+1]&&c1[i]==c2[j+1]) j++;
        if(j==m) pos[++c]=i-m+1,j=nxt[j];
    }
    return c;
}
inline int Get(char *s,lint *c,int m)
{
    int n=0;
    while(m--)
    {
        n++;scanf("%lld-%c",&c[n],&s[n]);
        if(n>1&&s[n]==s[n-1]) c[n-1]+=c[n],n--;
    }
    return n;
}
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    n=Get(s,cs,n),m=Get(t,ct,m);
    if(n<m) return !printf("0\n");
//  for(int i=1;i<=n;i++)
//      printf("%lld-%c ",cs[i],s[i]);
//  printf("\n");
//  for(int i=1;i<=m;i++)
//      printf("%lld-%c ",ct[i],t[i]);
//  printf("\n");
    if(m==1)
    {
        lint ans=0;
        for(int i=1;i<=n;i++)
            if(s[i]==t[1]) ans+=max(0ll,cs[i]-ct[1]+1);
        printf("%lld\n",ans);return 0;
    }
    else if(m==2)
    {
        int ans=0;
        for(int i=1;i<n;i++)
            if(s[i]==t[1]&&s[i+1]==t[2])
                ans+=(cs[i]>=ct[1]&&cs[i+1]>=ct[2]);
        printf("%d\n",ans);return 0;
    }
    int c,ans=0;get_nxt(t+1,ct+1,m-2);
    c=KMP(s,cs,t+1,ct+1,n,m-2);
    for(int i=1;i<=c;i++)
    {
        int x=pos[i]-1,y=pos[i]-1+m-1;
        if(!x||y>n) continue;
        if(s[x]!=t[1]||s[y]!=t[m]) continue;
        if(cs[x]>=ct[1]&&cs[y]>=ct[m]) ans++;
    }
    printf("%d\n",ans);return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值