KMP算法,需要注意的是题目要求的是int匹配,而非char匹配。
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <complex>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <cassert>
using namespace std;
int s[2000005],p[2000005],next1[2000005];
void getnext(int *p,int plen)
{
int i=0,j=-1;
next1[0]=-1;
while(i<plen)
{
if(j==-1||p[i]==p[j])
{
i++;
j++;
next1[i]=j;
}
else
j=next1[j];
}
}
bool boolkmp(int *s,int slen,int *p,int plen)
{
int i=0,j=0;
while(i<slen&&j<plen)
{
if(j==-1||s[i]==p[j])
{
i++;
j++;
}
else
j=next1[j];
}
if(j==plen)
return true;
else
return false;
}
int intkmp(int *s,int slen,int *p,int plen)
{
int i=0,j=0;
int times=0;
while(i<slen)
{
if(j==-1||s[i]==p[j])
{
i++;
j++;
}
else
j=next1[j];
if(j==plen)
{
times++;
j=next1[j];
}
}
return times;
}
int main()
{
int N,M,L,R;
cin>>N;
for(int i=0;i<N;i++) scanf("%d",&s[i]);;
cin>>M;
for(int i=0;i<M;i++) scanf("%d",&p[i]);
L=N-1;R=M-1;
for(int i=0;i<L;i++) s[i]=s[i+1]-s[i];
for(int i=0;i<R;i++) p[i]=p[i+1]-p[i];
getnext(p,R);
if(boolkmp(s,L,p,R)) cout<<"Wow! Life Winner!"<<'\n'<<intkmp(s,L,p,R)<<endl;
else cout<<"Oh. That's impossible. I must have had a dream."<<endl;
return 0;
}