KMP（SOJ3596）

(2012-07-15 19:28:27)

SOJ3596: http://cstest.scu.edu.cn/soj/problem.action?id=3596

这道题采用的算法是DP+KMP，也是我做KMP的题中最具挑战性的一道。

首先利用KMP算法将所有的word在text中的所有位置（包括起点和终点）存到一个struct数组中。

至于DP，设xiaoye[i]表示以text[i]为结尾的结果，则求xiaoye[i]即是对所有的满足：j<i，xiaoye[j]>0,并且从字符text[j+1]到字符text[i]是某一个word条件的xiaoye[j]作xiaoye[i]=xiaoye[i]+xiaoye[j]的处理。但是直接按照这个思路处理会超时，倒过来进行就会大大提高效率。

这需要对struct数组进行预处理，也就是下面的cmp函数。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int next[1005];
char text[1000005];
int sum;
struct node
{
    char pattern[1005];
}word[1005];
struct Node
{
    int a;
    int b;
}position[1000005];
void get_next(char *pattern)
{
    int j=0;
    int k=-1;
    next[0]=-1;
    while(pattern[j])
    {
        if(k==-1 || pattern[j]==pattern[k])
        {
            ++j;
            ++k;
            if(pattern[j]!=pattern[k])
                next[j]=k;
            else
                next[j]=next[k];
        }
        else
            k=next[k];
    }
}
void KMP(char *pattern)
{
    int i=0;
    int j=0;
    int result=0;
    int lP=strlen(pattern);
    while(text[i])
    {
        if(text[i]==pattern[j])
        {
            i++;
            j++;
        }
        else if(next[j]==-1)
        {
            j=0;
            i++;
        }
        else
            j=next[j];
        if(j==lP-1)
        {
            position[sum].a=i-lP+1;
            position[sum++].b=i-1;
            j=next[j];//这里的next[j]即是next['0']的值，显然只能>=0.
            
        }
    }
}
int cmp(Node x,Node y)
{
    return x.a<y.a || (x.a==y.a && x.b<y.b);
}
int xiaoye[1000005];
int main()
{
    int test;
    int n;
    int i;
    scanf("%d",&test);
    while(test--)
    {    
        sum=0;
        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            scanf("%s",word[i].pattern);
            strcat(word[i].pattern,"0");
        }
        scanf("%s",text);
        for(i=0;i<n;i++)
        {
            get_next(word[i].pattern);
            KMP(word[i].pattern);
        }
        int lT=strlen(text);
        memset(xiaoye,0,sizeof(xiaoye));
        sort(position,position+sum,cmp);
        for(i=0;i<sum;i++)
        {
            if(position[i].a==0)
                xiaoye[position[i].b]++;
            else
            {
                xiaoye[position[i].b]=(xiaoye[position[i].b]+xiaoye[position[i].a-1])%835672545;
            }
        }
        printf("%d\n",xiaoye[lT-1]);
    }
    return 0;
}