KMP模板

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int n;
char s[4005][205];
int nex[205];
char c[205];
void get_nex(char s[])
{
    int j=-1;
    nex[0]=-1;
    int len=strlen(s);
    for(int i=0;i<len;i++)
    {
        while(j!=-1&&s[j+1]!=s[i]) j=nex[j];
        if(s[i]==s[j+1]&&i!=0)   j++;
        nex[i]=j;
    }
}
bool KMP(char s[],char t[])
{            
    int j=-1;
    int len=strlen(s);
    int ok=strlen(t);
    for(int i=0;i<len;i++)
    {   
        while(j!=-1&&t[j+1]!=s[i])
        j=nex[j];
        if(s[i]==t[j+1])    //!!!
        j++;
        if(j==ok-1) return true;
    }
    return false;
}

上道裸题： 【POJ 3461】Oulipo ；

B中找几个A：

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
using namespace std;
int n;
int ans=0;
int nex[10005];
char s[10005],t[1000005];
inline void get_nex(char s[])
{
    int j=-1;
    int len=strlen(s);
    for(int i=0;i<len;i++)
    {
        while(j!=-1&&s[i]!=s[j+1]) j=nex[j];
        if(i!=0&&s[i]==s[j+1]) j++;
        nex[i]=j; 
    } 
} 
inline void KMP(char s[],char t[])
{
    int j=-1;
    int ok=strlen(s);
    int len=strlen(t);
    for(int i=0;i<len;i++)
    {
        while(j!=-1&&s[j+1]!=t[i]) j=nex[j];
        if(s[j+1]==t[i]) j++;
        if(j==ok-1) ans++,j=nex[j]; 
    }
}
int main()
{
scanf("%d",&n);
    while(n--)
    {
        ans=0;
        scanf("%s%s",s,t);
        get_nex(s);
        KMP(s,t);
        printf("%d\n",ans);
    }
}