KMP算法

很简单的字符串匹配，KMP以及它的一个简单优化。整理成板子。

#include<bits/stdc++.h> 
#define f(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
char s1[100005];
char s2[100005];
int nxt[100005];
//void getnxt(char s[],int nxt[]){
//    int len=strlen(s);
//    nxt[0]=-1;
//    int i=0,j=-1;
//    while(i<len-1){
//        if(j==-1||s[i]==s[j]){
//            i++,j++;
//            nxt[i]=j;
//        }
//        else j=nxt[j]; 
//    }
//}
void getnxt(char s[],int nxt[]){
    int len=strlen(s);
    nxt[0]=-1;
    int i=0,j=-1;
    while(i<len-1){
        if(j==-1||s[i]==s[j]){
            i++,j++;
            if(s[i]==s[j]) nxt[i]=nxt[j];
            else nxt[i]=j;
        }
        else j=nxt[j]; 
    }
}
int main(){
    while(~scanf("%s",s1)){
        if(s1[0]=='#') return 0;
        scanf("%s",s2);
        getnxt(s2,nxt);
        int len1=strlen(s1);
        int len2=strlen(s2);
        int k=0;
        int cnt=0;
        int i=0;
        while(i<=len1){
            while(k<len2&&i<=len1){
                if(k==-1||s1[i]==s2[k]){
                    i++,k++;
                }
                else{
                    k=nxt[k];
                }
            }
            if(k==len2){
                cnt++;
                k=0;
            }
        }
        printf("%d
",cnt);
    }
}

 上面是求有多少个独立子串，下面还有一个输出前缀数组和可不独立子串首次位置的代码。

#include<bits/stdc++.h> 
#define pf printf
#define f(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
char s1[100005];
char s2[100005];
int nxt[100005];
void getnxt(char s[],int nxt[]){
    int len=strlen(s);
    nxt[0]=-1;
    int i=0,j=-1;
    while(i<=len-1){
        if(j==-1||s[i]==s[j]){
            i++,j++;
            nxt[i]=j;
        }
        else j=nxt[j]; 
    }
}
//void getnxt(char s[],int nxt[]){
//    int len=strlen(s);
//    nxt[0]=-1;
//    int i=0,j=-1;
//    while(i<len-1){
//        if(j==-1||s[i]==s[j]){
//            i++,j++;
//            if(s[i]==s[j]) nxt[i]=nxt[j];
//            else nxt[i]=j;
//        }
//        else j=nxt[j]; 
//    }
//}
int main(){
    while(~scanf("%s",s1)){
        scanf("%s",s2);
        getnxt(s2,nxt);
        int len1=strlen(s1);
        int len2=strlen(s2);
        int k=0;
        int cnt=0;
        int i=0;
        while(i<=len1){
            while(k<len2&&i<=len1){
                if(k==-1||s1[i]==s2[k]){
                    i++,k++;
                }
                else{
                    k=nxt[k];
                }
            }
            if(k==len2){
                cnt++;
                k=nxt[len2-1];
                pf("%d
",i-len2+1);
                i--;
            }
        }
        printf("%d",nxt[1]);
        f(i,2,len2) printf(" %d",nxt[i]);
    }
}