p4824 kmp算法的运用

kmp算法的灵活运用

#include <stdio.h>
#include <string.h>
#include <algorithm>

using namespace std;
const int maxn=1e6+7;
char s[maxn],t[maxn];
int nxt[maxn];
int stk[maxn],top,p[maxn];
int n,m;

int main()
{
    scanf("%s%s",s+1,t+1);n=strlen(s+1),m=strlen(t+1);
    for(int i=2,j=0;i<=m;i++){
        while(j&&t[i]!=t[j+1])j=nxt[j];
        if(t[i]==t[j+1])j++;
        nxt[i]=j;
    }
    for(int i=1,j=0;i<=n;i++){
        while(j&&s[i]!=t[j+1])j=nxt[j];
        if(s[i]==t[j+1])j++;
        p[i]=j;
        stk[++top]=i;
        if(j==m){
            top-=m;
            j=p[stk[top]];
        }
    }
    for(int i=1;i<=top;i++)printf("%c",s[stk[i]]);printf("\n");
    return 0;
}