HDU--2203（kmp,模板题）

题目连接：

http://acm.hdu.edu.cn/showproblem.php?pid=2203

依旧是模板……循环移位了一遍还没包含那就永远不可能包含了吧

代码：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#define INF 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxd=1000000+5;
const int maxn=10000+5;
typedef long long ll;
typedef pair<int,int> pii;
//---------------------------
int n,m,ans;
string a,b;
vector<int> pos;


bool find_substring(string a,string b)
{
    vector<int> next(m+1,0);
    for(int i=1; i<m; ++i)
    {
        int j=i;
        while(j>0)
        {
            j=next[j];
            if(b[j]==b[i])
            {
                next[i+1]=j+1;
                break;
            }
        }
    }
    int i,j,cnt=0;
    for(i=0,j=0; cnt<2; ++i)
    {
        if(i==n) i=0,++cnt;
        if(j<m && a[i]==b[j])
            ++j;
        else
        {
            while(j>0)
            {
                j=next[j];
                if(a[i]==b[j])
                {
                    ++j;
                    break;
                }
            }
        }
        if(j==m) return true;
    }
    return false;
}

int main()
{
    freopen("in.txt", "r", stdin);
    while(cin>>a>>b)
    {
        n=a.length(),m=b.length();
        if(find_substring(a,b))
            printf("%s\n","yes");
        else
            printf("%s\n","no");

    }

    return 0;
}