D.截取方案数

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

#include <stdio.h>
#include <string.h>
const int N =  100000 + 10;
char s1[N],s2[N];
int next[N];
int len1,len2;
void getNext(){
    int i=1,j=0;
    next[1]=0;
    while(i<=len1){//优化后的next函数
        if(j==0||s2[i]==s2[j]){
            i++;j++;
            if(s2[i]!=s2[j]) next[i]=j;
            else next[i]=next[j];
        }
        else j=next[j];
    }
}


/*优化前的next函数
void getnext(){
int i = 1, j = 0;
next[1] = 0;
while(i <= len1){
if(j == 0 || s2[i] == s2[j]){
i++, j++;
next[i] = j;
}else{
j = next[j];
}
}
}*/
int kmp(){
    int i,j;
    int res=0;
    for(i=1,j=0; i<=len1; i++)
    {
        while(j>0&&s2[j+1]!=s1[i])
         j=next[j];
        if(s2[j+1] == s1[i])
            j= j + 1;
        if(j==len2)
        {
            res++;
            j=0;
        }
    }
    return res;
}
int main(){
    int i,j,res;
    while(gets(s2+1)){
        gets(s1+1);
        len1=strlen(s1+1);
        len2=strlen(s2+1);
        getNext();
        printf("%d\n",kmp());
    }
    return 0;
}