KMP

KMP

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int Max=1e6+100;
char a[Max],b[Max];
int x1,x2,nxt[Max];
void getnxt()
{
    nxt[0]=-1;
    int k=-1,j=0;
    while(j<x2)
    {
        if(k==-1||b[k]==b[j])//模式串
        {
            nxt[++j]=++k;
            if(b[j]!=b[k])
                nxt[j]=k;
            else nxt[j]=nxt[k];
        }
        else k=nxt[k];
    }
}
int kmp()
{
    int i=0,j=0;
    if(x1<x2)//x1是文本串，x2是模式串
        return -1;
    getnxt();
    while(i<x1&&j<x2)
    {
        if(j==-1||a[i]==b[j])
            i++,j++;
        else j=nxt[j];
    }
    if(j==x2)
        return i-j;//返回模式串在文本串中对应的下标
    else return -1;//找不到
}
int main()
{
    scanf("%s",a);
    x1=strlen(a);
    scanf("%s",b);
    x2=strlen(b);
    int x=kmp();
    printf("%d\n",x);
    return 0;
}