【poj3461】kmp裸题

poj3461

直接看样例：T组数据，每组包括两个字符串AB，求A串在B串所有匹配位置的总数。

KMP裸题……那天刚学KMP于是拿来练手。

【想看代码的直接拉下去】

KMP是一个字符串匹配的算法，可以在O(n)的时间内求出一个串在另一个串的所有匹配位置。
暴力打法是在检查的时候，如果失配则往后移一位再进行匹配…KMP算法认为这样浪费了一些信息。因为若在第k位失配，我们已经知道[0,k-1]位的信息，却只往后移一位进行匹配，显然是很浪费的。
比如匹配abcba时在第四位失配，那么我们已经知道那几位是abcb，然后拿a和b来匹配，这个匹配显然是错误的。

所以说，我们可以用一个nxt数组表示第I位失配后跳到nxt[i]位。
而nxt数组的求法则是自己和自己匹配。

其实KMP主要就是背代码，基本上没法拓展了…我背的大白上的。

下面是代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;


int nxt[1000010];

void getnxt(char *a)
{
    memset(nxt,0,sizeof(nxt));
    int l=strlen(a);
    nxt[1]=nxt[0]=0;
    for(int i=1;i<l;i++)
    {
        int j=nxt[i];
        while(j&&a[i]!=a[j]) j=nxt[j];
        nxt[i+1]= a[i]==a[j]? j+1 :0;
    }
}

int kmp(char *a,char *b)
{
    int n=strlen(a),m=strlen(b);
    getnxt(b);
    int ans=0;
    for(int i=0,j=0;i<n;i++)
    {
        while(j&&a[i]!=b[j]) j=nxt[j];
        if(a[i]==b[j]) j++;
        if(j==m) ans++;
    }
    return ans;
}

char a[1000010],b[1000010];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s%s",a,b);
        printf("%d\n",kmp(b,a));
    }
    return 0;
}

---

2016.2.24

今天又来敲了一遍…这东西敲的少啊QAQ

还有就是有个新发现，strlen()好像不是O(1)的…别问我怎么知道的，T了两次…

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int SZ = 1000010;

int nxt[SZ];

void getnxt(char P[])
{
    memset(nxt,0,sizeof(nxt));
    nxt[0] = nxt[1] = 0;
    int n = strlen(P);
    for(int i = 1;i < n;i ++)
    {
        int j = nxt[i];
        while(j && P[i] != P[j]) j = nxt[j];
        nxt[i + 1] = P[i] == P[j] ? j + 1 : 0;
    }
}

int ask(char P[],char T[])
{
    getnxt(P);
    int ans = 0;
    int m = strlen(P),n = strlen(T);
    for(int i = 0,j = 0;i < n;i ++)
    {
        while(j && P[j] != T[i]) j = nxt[j];
        if(P[j] == T[i]) j ++;
        if(j == m) ans ++;
    }
    return ans;
}

char s1[SZ],s2[SZ];

int main()
{
    int T;
    scanf("%d",&T);
    while(T --)
    {
        scanf("%s%s",s1,s2);
        printf("%d\n",ask(s1,s2));
    }
    return 0;
}