【KMP】【CodeForces】contest 1016 B Segment Occurrences

【题目】http://codeforces.com/contest/1016/problem/B

【题意】给定两个字符串，若干次询问，问在询问区间内，a中有多少个b

【思路】KMP

【模板】https://www.cnblogs.com/chenxiwenruo/p/3546457.html

【代码】

​#include


int next[10005];
int book[10005]= {0};


//求str对应的next数组
void getNext(char const* str, int len)
{
    int i = 0;
    next[i] = -1;
    int j = -1;
    while( i < len )
    {
        if( j == -1 || str[i] == str[j] )   //循环的if部分
        {
            ++i;
            ++j;
            //修正的地方就发生下面这4行
            if( str[i] != str[j] ) //++i，++j之后，再次判断ptrn[i]与ptrn[j]的关系
                next[i] = j;      //之前的错误解法就在于整个判断只有这一句。
            else
                next[i] = next[j];  //这里其实是优化了后的，也可以仍是next[i]=j
            //当str[i]==str[j]时，如果str[i]匹配失败，那么换成str[j]肯定也匹配失败，
            //所以不是令next[i]=j，而是next[i] = next[j]，跳过了第j个字符，
            //即省去了不必要的比较
            //非优化前的next[i]表示前i个字符中前缀与后缀相同的最大长度
        }
        else                                 //循环的else部分
            j = next[j];
    }
}

//在目标字符串target中，字符str出现的个数
//n为target字符串的长度，m为str字符串的长度
int kmp_match(char *target,int n,char *str,int m)
{
    int i=0,j=0;  //i为target中字符的下标，j为str中字符的下标
    int cnt=0;   //统计str字符串在target字符串中出现的次数
    while(i<=n-1)
    {
        if(j<0||target[i]==str[j])
        {
            i++;
            j++;
        }
        else
        {
            j=next[j]; //当j=0的时候，suffix[0]=-1，这样j就会小于0，所以一开始有判断j是否小于0
        }

        //str在target中找到匹配
        if(j==m)
        {
            cnt++;
            book[i-1]=1;
            j=next[j];
        }
    }
    return cnt;
}

int main()
{
    int n,m,q;
    char a[10005],b[10005];
    scanf("%d%d%d",&n,&m,&q);
    scanf("%s",a);
    scanf("%s",b);
    int flag=0;
    if(strlen(b)>strlen(a))
        flag=1;
    getNext(b,strlen(b));
    kmp_match(a,strlen(a),b,strlen(b));
    for(int i=0; i