HDU 2087 / POJ 3461 KMP拓展

hdu 2087 较为容易的kmp拓展。

一旦模式串匹配成功，匹配成功之后，直接从后继续比较。

#include<cstdio>
#include<cstring>
const int maxn=1005;
char str1[maxn],str2[maxn];
int next[maxn];

void getnext(char *str,int len)
{
    next[1]=0;
    int k=0;

    for(int i=2;i<len+1;i++)
    {
        while(k>0&&str[k]!=str[i-1]) k=next[k];
        if(str[k]==str[i-1]) k++;
        next[i]=k;
    }
}

int kmp(char *str1,char *str2,int len1,int len2)
{
    getnext(str2,len2);
    int ans=0;

    int k=0;
    for(int i=0;i<len1;i++)
    {
        while(k>0&&str1[i]!=str2[k]) k=next[k];
        if(str1[i]==str2[k]) k++;
        if(k==len2) {k=0; ans++;}
    }
    return ans;
}

int main()
{
    while(~scanf("%s",str1)&&strcmp(str1,"#")!=0)
    {
        scanf("%s",str2);
        int len1=strlen(str1),len2=strlen(str2);
        printf("%d\n",kmp(str1,str2,len1,len2));
    }
    return 0;
}

POJ 3461

跟hdu2087有点像。

匹配成功之后不是简单地向后继续匹配。

通过next数组获得结尾字符的循环度之后，平移模式串，对齐，匹配。

比较奇怪，此代码G++ WA，C++ AC。

#include<cstdio>
#include<cstring>
const int maxn=1000000;
char str1[maxn],str2[maxn];
int next[maxn];

void getnext(char *str,int len)
{
    next[1]=0;
    int k=0;

    for(int i=2;i<len+1;i++)
    {
        while(k>0&&str[k]!=str[i-1]) k=next[k];
        if(str[k]==str[i-1]) k++;
        next[i]=k;
    }
}

int kmp(char *str1,char *str2,int len1,int len2)
{
    getnext(str2,len2);
    int ans=0;

    int k=0;
    for(int i=0;i<len1;i++)
    {
        while(k>0&&str1[i]!=str2[k]) k=next[k];
        if(str1[i]==str2[k]) k++;
        if(k==len2) {k=next[k]; ans++;}
    }
    return ans;
}

int main()
{
    int t;
    scanf("%d",&t);
    getchar();
    while(t--)
    {
        scanf("%s%s",str2,str1);
        int len1=strlen(str1),len2=strlen(str2);
        printf("%d\n",kmp(str1,str2,len1,len2));
    }
    return 0;
}