KMP算法（练习题）

hdu 1711 Number Sequence

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int t,n,m,a[N],b[N],ne[N];//不要定义成next[N]，杭电oj的G++编译过不了，有变量名字冲突
void get_next()
{
    memset(ne,0,sizeof(ne));
    int i=1,j=0;
    ne[1]=0;
    while(i<=m)//打表打到next[m+1]
    {
        if(j==0||b[i]==b[j])i++,j++,ne[i]=j;
        else j=ne[j];
    }
}
int kmp()//匹配后，子串第一个字符在主串上的最小位置
{
    int i=1,j=1;
    while(i<=n&&j<=m)
    {
        if(j==0||a[i]==b[j])i++,j++;
        else j=ne[j];
        if(j==m+1)return i-m;
    }
    return -1;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        for(int i=1;i<=m;i++)
            cin>>b[i];
        get_next();
        printf("%d\n",kmp());
    }
    return 0;
}

hdu 1686 Oulipo

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
char a[N],b[N];
int t,n,m,ne[N];
void get_next()
{
    memset(ne,0,sizeof(ne));
    int i=1,j=0;
    ne[1]=0;
    while(i<=m)//打表打到next[m+1]
    {
        if(j==0||b[i]==b[j])i++,j++,ne[i]=j;
        else j=ne[j];
    }
}
int kmp()//返回匹配成功的次数
{
    int i=1,j=1,s=0;
    while(i<=n&&j<=m)
    {
        if(j==0||a[i]==b[j])i++,j++;
        else j=ne[j];
        if(j==m+1)s++,j=ne[j];//注意要滑动j
    }
    return s;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--)
    {
        cin>>b+1>>a+1;
        n=strlen(a+1);
        m=strlen(b+1);
        get_next();
        printf("%d\n",kmp());
    }
    return 0;
}

洛谷 P3375 【模板】KMP字符串匹配

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
char a[N],b[N];
int n,m,next[N];
void get_next()
{
    int j=0;
    next[1]=0;//next[1]是确定值，为0
    for(int i=2;i<=n;i++)//i从2开始
    {
        while(j!=0&&b[i]!=b[j+1])j=next[j];
        if(b[i]==b[j+1])j++;
        next[i]=j;
    }
}
void kmp()
{
    int j=0;
    for(int i=1;i<=n;i++)
    {
        while(j!=0&&a[i]!=b[j+1])j=next[j];
        if(a[i]==b[j+1])j++;
        if(j==m){printf("%d\n",i-m+1);j=next[j];}
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>a+1>>b+1;//地址加1，即字符串下标从1开始
    n=strlen(a+1);//主串长度为n
    m=strlen(b+1);//子串长度为m
    get_next();kmp();
    for(int i=1;i<=m;i++)
        i==m?printf("%d\n",next[i]):printf("%d ",next[i]);
    return 0;
}