hdu 2594 擴展KMP模板題

//M hdu2594 擴展KMP  模板題
//題意：給定兩個字符串，在第一個字符串中找到一個最大前綴作爲第二個字符串的後綴
//思路：將S1作爲模式串 然後在s2中尋找，S2作爲目標串，用KMP就好，看到S2的最後一個字符時能匹配S1中的第幾個字符；

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pp;
#define inf 0x3f3f3f3f
#define eps 1e-10
#define maxl 50050
#define mem(i,j) memset(i,j,sizeof(i))
const int mod=1e9+7;
char s1[maxl],s2[maxl];
int f[maxl],len1,len2;

void getnext(char *s,int *f){
    memset(f,0,sizeof(f));
    for(int i=1;i<len1;i++){
        int j=f[i];
        while(j&&s[i]!=s[j]) j=f[j];//這裏不斷的替換來降低複雜度
        f[i+1]=(s[i]==s[j])?j+1:0;//f[i+1]保存的是s[i]的前綴的長度
    }
}

int extendKMP(){
    int j=0;
    for(int i=0;i<len2;i++){
        while(j&&s2[i]!=s1[j]) j=f[j];//這裏是爲了下次匹配的時候不再是j==0開始
        if(s2[i]==s1[j]) j++;
    }
    return j;
}

int main()
{
    freopen("in.txt", "r", stdin);
    while(scanf("%s%s",s1,s2)!=EOF){
        len1=strlen(s1);
        len2=strlen(s2);
        getnext(s1,f);
        int len= extendKMP();
        for(int i=0;i<len;i++) {
            if(i!=len-1)cout<<s1[i];
            else cout<<s1[i]<<' ';
        }
        cout<<len<<endl;
    }
    return 0;
}