最近看了看KMP算法,发现其状态转移图构造的十分巧妙,通过递推计算出返回的地方,以降低时间复杂度。 该题就是一个KMP裸题,不过该题可以循环 移位,那么不妨我们将s1扩充成原来的两倍就行了,但是这样要实现判断两个串的大小,不然扩充后的s1有可能匹配上一个比它长的s2 。
细节参见代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int maxn = 100000 + 5;
int T,n,m,f[maxn];
char s1[2*maxn],s2[maxn];
void getFail(char* P, int* f) {
int m = strlen(P);
f[0] = 0; f[1] = 0;
for(int i = 1; i < m; i++) {
int j = f[i];
while(j && P[i]!=P[j]) j = f[j];
f[i+1] = P[i] == P[j] ? j+1 : 0;
}
}
bool findd(char* T, char* P, int* f) {
int n = strlen(T), m = strlen(P);
int nn = n;
for(int i=0;i<n;i++)
T[nn++] = T[i];
n = nn;
getFail(P, f);
int j = 0;
bool ok = false;
for(int i=0;i<n;i++) {
while(j && P[j]!=T[i]) j = f[j];
if(P[j] == T[i]) j++;
if(j == m) { ok = true; break; }
}
return ok;
}
int main() {
while(~scanf("%s",s1)) {
scanf("%s",s2);
if(strlen(s1) < strlen(s2)) printf("no\n");
else if(findd(s1,s2,f)) printf("yes\n");
else printf("no\n");
}
return 0;
}