义冢OJ P5018 Oulipo

描述

给出两个字符串W和T,求T中有几个W子串

輸入

第一行为数据组数.

每组数据有两行W和T,表示模式串和原始串.

輸出

对每组数据,每行一个数,表示匹配数.

輸入範例 1

3
BAPC
BAPC
AZA
AZAZAZA
VERDI
AVERDXIVYERDIAN

輸出範例 1

1
3
0

提示

1 ≤ |W| ≤ 10,000 (here |W| denotes the length of the string W). |W| ≤ |T| ≤ 1,000,000.

KMP板子题

只能写写板子来水水博客了。。。。

Code

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

const int MAXN=1e6+5;

int nxt[MAXN]={0};

void setNext(string s);
int kmp(string s1,string s2);

int n;

int main()
{
    cin>>n;
    string s1,s2;
    for(int i=1;i<=n;++i)
    {
        cin>>s1>>s2;
        setNext(s1);
        cout<<kmp(s2,s1)<<endl;
    }
    
    return 0;
}
void setNext(string s)
{
    memset(nxt,0,sizeof(nxt));
    nxt[0]=-1;
    int sl=s.size();
    for(int i=1;i<sl;++i)
    {
        int t=nxt[i-1];
        while(s[t+1]!=s[i]&&t>=0)
        {
            t=nxt[t];
        }
        if(s[t+1]==s[i])
        {
            nxt[i]=t+1;
        }
        else
        {
            nxt[i]=-1;
        }
    }
}
int kmp(string s1,string s2)
{
    int i=0,j=0;
    int sl1=s1.size(),sl2=s2.size();
    int cnt=0;
    while(i<sl1)
    {
        if(s1[i]==s2[j])
        {
            ++i;
            ++j;
            if(j==sl2)
            {
                ++cnt;
                j=nxt[j-1]+1;
            }
        }
        else
        {
            if(j==0)++i;
            else j=nxt[j-1]+1;
        }
    }
    return cnt;
}