codeforces 149E Martian Strings(KMP)

题目链接：http://codeforces.com/problemset/problem/149/E

题意：给定一个串S和一个串T，在S找出四个位置a,b,c,d(a<=b<c<=d)，使得S[a,b]+S[c,d]=T？

思路：S和T匹配一次，记录T的位置i能匹配到S的最小位置；翻转S和T再匹配一次。

char s1[N],s2[N],p1[N],p2[N];

int n,Next1[N],Next2[N];



void reverse(char *s,int len)

{

    int L=0,R=len-1;

    while(L<R) swap(s[L++],s[R--]);

}



void getNext(char s[],int len,int next[])

{

    next[0]=-1;

    int i=0,j=-1;

    while(i<len)

    {

        if(j==-1||s[i]==s[j]) next[++i]=++j;

        else j=next[j];

    }

}





int pos[2][N];



void match(char s[],int sLen,char p[],int pLen,int next[],int pos[])

{

    int i=0,j=0;

    while(i<sLen)

    {

        if(j==-1||s[i]==p[j])

        {

            i++;j++;

            pos[j]=min(pos[j],i);

        }

        else j=next[j];

        if(j==pLen) j=next[j];

    }

}



int main()

{

    RD(s1);

    int len=strlen(s1);

    strcpy(s2,s1); reverse(s2,len);

    int ans=0,i,j,L;

    RD(n);

    while(n--)

    {

        RD(p1); strcpy(p2,p1); L=strlen(p1); reverse(p2,L);

        if(L<=1) continue;

        FOR1(i,L) pos[0][i]=pos[1][i]=INF;

        getNext(p1,L,Next1);

        match(s1,len,p1,L,Next1,pos[0]);



        getNext(p2,L,Next2);

        match(s2,len,p2,L,Next2,pos[1]);

        FOR1(i,L) if(pos[0][i]+pos[1][L-i]<=len)

        {

            ans++;

            break;

        }

    }

    PR(ans);

    return 0;

}