HDU_2203_KMP入门

亲和串

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 12533    Accepted Submission(s): 5667

Problem Description

人随着岁数的增长是越大越聪明还是越大越笨，这是一个值得全世界科学家思考的问题,同样的问题Eddy也一直在思考，因为他在很小的时候就知道亲和串如何判断了，但是发现，现在长大了却不知道怎么去判断亲和串了，于是他只好又再一次来请教聪明且乐于助人的你来解决这个问题。
亲和串的定义是这样的：给定两个字符串s1和s2,如果能通过s1循环移位，使s2包含在s1中，那么我们就说s2 是s1的亲和串。

 

Input

本题有多组测试数据，每组数据的第一行包含输入字符串s1,第二行包含输入字符串s2，s1与s2的长度均小于100000。

Output

如果s2是s1的亲和串，则输出"yes"，反之，输出"no"。每组测试的输出占一行。

 

Sample Input

AABCD

CDAA

ASD

ASDF

 

Sample Output

yes

no

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<algorithm>
#include<cmath>
#include<string>
using namespace std;

char a[200005],b[100005];
int Next[100005];

void getNext()
{
    memset(Next,-1,sizeof(Next));  //！！！下标从0开始和从1开始，初始化数组不同，在这TLE了好久
    int m=strlen(b),k=-1;
    for(int i=1; i<m; i++)
    {
        while(k>-1&&b[k+1]!=b[i])
            k=Next[k];
        if(b[k+1]==b[i])
            k++;
        Next[i]=k;
    }
}

void CMP()
{
    getNext();
    int n=strlen(a),m=strlen(b),k=-1;
    for(int i=0; i<n; i++)
        a[i+n]=a[i];
    a[2*n]='';
    n<<=1;
    for(int i=1; i<n; i++)
    {
        while(k>-1&&b[k+1]!=a[i])
            k=Next[k];
        if(b[k+1]==a[i])
            k++;
        if(k==m-1)
        {
            puts("yes");
            return;
        }
    }
    puts("no");
}

int main()
{
    while(scanf("%s%s",a,b)!=EOF)
    {
        CMP();
    }
    return 0;
}