UVALive5876-Writings on the Wall-KMP

有两段字符串,第一段的尾和第二段的头可能重合。问有多少种组合的可能。

需要理解一下next数组的意义。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 100100;

int T;
int next[maxn];
char s1[maxn],s2[maxn],s[maxn];

int getnext(char T[],int m)
{
    int i=0,j=-1;
    next[0] = -1;
    while(i < m)
    {
        if(j == -1 || T[j] == T[i])
        {
            i++;j++;
            next[i] = j;
        }
        else
            j = next[j];
    }
}

int main()
{
    scanf("%d",&T);
    for(int i=0;i<T;i++)
    {
        scanf("%s%s",s1,s2);

        int l1 = strlen(s1),l2 = strlen(s2);
        for(int i=0;i<l2;i++) s[i] = s2[i];
        s[l2] = '~';
        for(int i=l2+1;i<l2+l1+1;i++) s[i] = s1[i-l2-1];
        s[l1+l2+1] = '\0';

        int  len = l1+l2+1;
        getnext(s,len);


        int ans = 0,p=len;
        while(next[p] != -1)
        {
            ans++;
            p = next[p];
        }
        printf("%d\n",ans);
    }
}

 

posted @ 2016-02-17 23:15  Helica  阅读(169)  评论(0编辑  收藏  举报