hrbust/哈理工oj 2060 截取方案数【KMP匹配】

截取方案数

Time Limit: 1000 MS Memory Limit: 32768 K Total Submit: 165(70 users) Total Accepted: 68(58 users) Rating:  Special Judge: No

Description

给定一个模式串T，主串S，问：从S中截取T有多少种方案？

Input

有多组测试数据，对于每组测试数据，第一行是模式串T，第二行是主串S，数据中仅包含大小写字母和数字，模式串T长度不超过10^4， 主串S长度不超过10^5。 注意：数据是随机的。

Output

对于每组测试数据，输出一行，为截取方案数。

Sample Input

abc abcdaabcab abcd abcdaabcab aba abababa

Sample Output

2 1 3

Source

HCPC2014校赛训练赛 3

基础KMP匹配，如果匹配成功output++即可。

#include<stdio.h>
#include<string.h>
using namespace std;
int next[200005];
char a[2000005];
char b[2000005];
int lena;
int lenb;
int output;
void set_naxt()//子串的next数组
{
int i=0,j=-1;
next[0]=-1;
while(i<lenb)
{
if(j==-1||b[i]==b[j])
{
i++; j++;
next[i]=j;
}
else
j=next[j];
}
}

int kmp()
{
int i=0,j=0;
set_naxt();
while(i<lena)
{
if(j==-1||a[i]==b[j])
{
i++;j++;
}
else
j=next[j];
if(j==lenb)
{
output++;
}
}
}
int main()
{
while(~scanf("%s",b))
{
output=0;
scanf("%s",a);
lena=strlen(a);
lenb=strlen(b);
kmp();
printf("%d\n",output);
}
}