[Hiho]1015-KMP算法

题目：
输入

第一行一个整数N，表示测试数据组数。

接下来的N*2行，每两行表示一个测试数据。在每一个测试数据中，第一行为模式串，由不超过10^4个大写字母组成，第二行为原串，由不超过10^6个大写字母组成。

其中N<=20
输出

对于每一个测试数据，按照它们在输入中出现的顺序输出一行Ans，表示模式串在原串中出现的次数。
Sample Input

5
HA
HAHAHA
WQN
WQN
ADA
ADADADA
BABABB
BABABABABABABABABB
DAD
ADDAADAADDAAADAAD

Sample Output

3
1
3
1
0

Solution:
需要稍加改造KMP算法，因为有重叠的现象，所以需要将模式串的next数组往后再计算一位，比如ADA,首尾是一样的，那就计算ADAX，X所在next数，这样重叠现象也可解决。

#include<iostream>
#include<string>

using namespace std;
int * BuildNext(string pattern)
{
    int n = pattern.size();
    int* next = new int[n+1];
    int k = -1;
    next[0] = -1;
    int j = 0;
    while(j < n)
    {
        if(k == -1 || pattern[j] == pattern[k])
        {
            k++;
            j++;
            if( j == n || pattern[j] != pattern[k])
            {
                next[j] = k;
            }
            else
                next[j] = next[k];
        }
        else
            k  = next[k];
    }

    return next;
}

int Search(string pattern,string original)
{

    int n = original.size();
    int *next = BuildNext(pattern);
    int j = 0;
    int count = 0;
    for(int i = 0;i<original.size();i++)
    {
        if(j > 0 && pattern[j] != original[i])
        {
            j = next[j];
            if(j == -1)
                 j = 0;
        }
        if(pattern[j] == original[i])
            j++;
        if(j == pattern.size())
        {
            count++;
            j = next[j];
            if(j == -1)
                j = 0;
        }
    }

    return count;
}


int main()
{
    char rows;
    int row = 0;
    int i,j,k;
    i = j = k = 0;
    cin>>row;
    string* pattern = new string[row];
    string* original = new string[row];
    for(i = 0;i<row;i++)
    {
            cin>>pattern[i];
            cin>>original[i];
    }
    for(i = 0;i<row;i++)
    {
        int count  = Search(pattern[i],original[i]);
        cout<<count<<endl;
    }
    return 0;
}