hdu 亲和串(kmp)

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

 

Author

Eddy

 

Recommend

lcy   |   We have carefully selected several similar problems for you:  1358 3336 3746 2205 2207 

 

 

题解:注意循环...

 

代码:

#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <string>
#include <cstring>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f

char p[100010],s[100010];
int nex[100010];
int lens;

void get(char *p)
{
    int plen=strlen(p);
    nex[0]=-1;
    int k=-1,j=0;
    while(j < plen){
        if(k==-1 || p[j] == p[k]){
            ++j;
            ++k;
            if(p[j] != p[k])
                nex[j]=k;
            else
                nex[j]=nex[k];
        }
        else{
            k=nex[k];
        }
    }
}

int kmp(int lens,char *p)
{
    int i=0,j=0;
    int slen=lens;
    int plen=strlen(p);
    while(i < slen && j< plen){
        if(j==-1 || s[i]==p[j]){
            ++i;
            ++j;
        }
        else{
            j=nex[j];
        }
    }
    if(j == plen)
        return i-j;
    else
        return -1;
}

int main()
{
    while(~scanf("%s%s",s,p)){
        lens=strlen(s);
        for(int i=lens; i<2*lens; i++){
            s[i]=s[i-lens];
        }
        get(p);
        lens+=lens;
        if(kmp(lens,p)!=-1)
            puts("yes");
        else
            puts("no");
    }
}