Problem Description

人隨著歲數的增長是越大越聰明還是越大越笨，這是一個值得全世界科學家思考的問題,同樣的問題Eddy也一直在思考，因為他在很小的時候就知道親和串如何判斷了，但是發現，現在長大了卻不知道怎麼去判斷親和串了，於是他隻好又再一次來請教聰明且樂於助人的你來解決這個問題。
親和串的定義是這樣的：給定兩個字符串s1和s2,如果能通過s1循環移位，使s2包含在s1中，那麼我們就說s2 是s1的親和串。

 

Input

本題有多組測試數據，每組數據的第一行包含輸入字符串s1,第二行包含輸入字符串s2，s1與s2的長度均小於100000。

 

Output

如果s2是s1的親和串，則輸出"yes"，反之，輸出"no"。每組測試的輸出占一行。

 

Sample Input

 

Sample Output

 

Author

Eddy

Recommend

lcy

 

所以這裏我有優化就是當母串小於子串 直接輸出 不做匹配 實際上確實是這樣 詳細見代碼

 

 

#include <iostream>
#include<cstring>
#include<cstdio>
using namespace std;

string s0,s1,s2;
int next[200005];
void get_next(const string sub, int *next)
{
    int len=sub.length();
    int i,k;
    next[0]=k=-1;
    for (i=0; i<len;)
    {
        if (k==-1 || sub[i]==sub[k])
        {
            k++;
            i++;
            if (sub[k]!=sub[i]) next[i]=k;
            else next[i]=next[k];    //避免重複計算優化next數組
        }
        else k=next[k];
    }
}
int KMP(const string str, const string sub, const int *next)    //返回子串在主串中的起始位置下標
{
    int i,j;
    int len1=str.length();
    int len2=sub.length();
    for (i=0, j=0; i<len1 && j<len2;)
    {
        if (j==-1 || str[i]==sub[j])
        {
            i++;
            j++;
        }
        else
        {
            j=next[j];
        }

    }
    if (j==len2) return i-len2;
    return -1;    //如果找不到就返回-1
}
int main()
{
    while(cin>>s0>>s2)
    {
        if(s2.length()>s0.length())
        {
            cout<<"no"<<endl;
        }
        else
        {
            s1=s0+s0;
            get_next(s2,next);
            int ans=KMP(s1,s2,next);
            // cout<<ans<<endl;
            if(ans==-1)
                cout<<"no"<<endl;
            else
                cout<<"yes"<<endl;
        }
    }
    return 0;
}