第3次CCF-3-字符串匹配（kmp的简单应用）

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<limits.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
#include<vector>
#include<math.h>
#include<string>
using namespace std;
#define maxn 105
char s[maxn];
char str[maxn][maxn];
char str1[maxn][maxn];
bool flag[maxn]={0};
int n,m;
int p[1005];
void print()
{
int i,j=0;
p[1]=0;
int len=strlen(s+1);
for(i=2;i<=len;i++)
{
while(j>0 && s[j+1]!=s[i])
j=p[j];
if(s[i]==s[j+1])
j++;
p[i]=j;
}
}
void kmp(int x)
{
int k=x,i,j=0,sum=strlen(s+1);
int len=strlen(str[k]+1);
for(i=1;i<=len;i++)
{
while(j>0 && str[k][i]!=s[j+1])
j=p[j];
if(s[j+1]==str[k][i])
j++;
if(j==sum)
{
flag[k]=1;
break;
}
}
}
void work1()
{
for(int i=1;s[i]!='\0';i++)
s[i]=tolower(s[i]);
for(int i=1;i<=n;i++)
{
scanf("%s",str[i]+1);
strcpy(str1[i]+1,str[i]+1);
for(int j=1;str[i][j]!='\0';j++)
str[i][j]=tolower(str[i][j]);
print();
kmp(i);
}

}
void work2()
{
for(int i=1;i<=n;i++)
{
scanf("%s",str[i]+1);
print();
kmp(i);
}
}
int  main()
{
int i,j;
scanf("%s",s+1);
scanf("%d%d",&m,&n);
if(m==0)
work1();
else
work2();
for(i=1;i<=n;i++)
{
if(m==0)
{
if(flag[i])
printf("%s\n",str1[i]+1);
}
else
{
if(flag[i])
printf("%s\n",str[i]+1);
}
}
}