KMP模板+习题

博客首文，弱存下KMP模板，之前一直记在本子上，今天翻了下大神带码，自己又打了一遍，顺便上几道例题。orz

附上链接：点击打开链接

#include <bits/stdc++.h>
int f[10000];
using namespace std;
void getfill(string s)
{
    memset(f,0,sizeof(f)); 
    for(int i=1;i<s.size();i++)
    {
        int j=f[i];
        while(j && s[i]!=s[j])
            j=f[j];
        f[i+1]=(s[i]==s[j])?j+1:0;
    }
}
int find(string a,string s)
{
    getfill(s);
    int j=0;
    for(int i=0;i<a.size();i++)
    {
        while(j&&a[i]!=s[j])
            j=f[j];
        if(a[i]==s[j])
            j++;
        if(j==s.size())
            return i-s.size()+1;
    }
}

int main()
{
    string a,s;
    string s1="You get it",s2="Cry Cry";
    while(cin>>a>>s)
    {
        int pos=find(a,s);
        cout << pos << endl;
        if(pos!=a.size())
            cout<< s1 <<endl;
        else
            cout << s2 <<endl;
    }

    return 0;
}

附上大神总结：KMP专题

1.HDU1711点击打开链接

裸题一发，不多解释；（然而，第一遍竟然错了）

#include <bits/stdc++.h>
//HDU1711
//AC1
int f[1000010];
int a[1000010],s[1000010],n,m;
using namespace std;
void getfill(int* s)
{
    memset(f,0,sizeof(f));
    for(int i=1;i<m;i++)
    {
        int j=f[i];
        while(j && s[i]!=s[j])
            j=f[j];
        f[i+1]=(s[i]==s[j])?j+1:0;
    }
}

void find(int* a,int* s)
{
    getfill(s);
    int j=0;
    for(int i=0;i<n;i++)
    {
        while(j&&a[i]!=s[j])
            j=f[j];
        if(a[i]==s[j])
            j++;
        if(j==m)
        {
            printf("%d\n",i-m+2);
            return;
        }
    }
    printf("-1\n");
    return;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);

        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&s[i]);
        }
        if(m>n)
        {
            cout << -1 <<endl;
            continue;
        }
        find(a,s);
    }
    return 0;
}

POJ2406 点击打开链接

题意：找出最长周期串；

解法：利用next数组的性质，N%(N-next[N-1]+1)==0,则子串长度为（N-next[N-1]+1）

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <string.h>
#include <string>
//poj2406
//next[]数组表示的是，前缀字符串相同的个数
//a b a b a b--T=2
//0 0 1 2 3 4
//a b c a b c--T=3
//0 0 0 1 2 3
//a b c d a b--T不存在
//0 0 0 0 1 2
//N%(N-next[N])==0
using namespace std;
int f[1000005];
void getfill(string s)
{
    memset(f,0,sizeof(f));
    for(int i=1; i<s.size(); i++)
    {
        int j=f[i];
        while(j&&s[i]!=s[i])
            j=f[j];
        f[i+1]=(s[i]==s[j])?j+1:0;
    }
}


int main()
{
    string s;
    while(cin>>s)
    {
        int i=0;
        if(s==".")
            break;
            getfill(s);
            int n=s.size();
            if(n%(n-f[n-1]-1)==0)
            {
                printf("%d\n",n/(n-f[n]));
            }
            else
            {
                printf("1\n");
            }
    }
    return 0;
}