KMP

http://hihocoder.com/problemset/problem/1015

AC:

#include<iostream>  
#include<cstring> 
#include<string>
using namespace std;  
int Next[10001];
//求KMP的Next数组  
void GetNext(const string& p)  
{  
    int len=p.size();  
    int i=0;  
    int j=-1;  
    Next[i]=j;  
    while(i<len)  
    {  
        if(j==-1||p.at(i)==p.at(j))  
        {  
            i++;  
            j++;  
            if(i==len||p.at(i)!=p.at(j))//不允许出现P[i]==P[next[i]]  
                Next[i]=j;  
            else  
                Next[i]=Next[j];  
        }  
        else  
        {  
            j=Next[j];  
        }  
    }  
      
} 
 
 
//S为母串，p为匹配子串，如果匹配返回匹配位置，否则返回-1  
int KMPSearch(const string& s,const string& p)  
{  
    unsigned int Slen=s.size();  
    unsigned int Plen=p.size();
    //int *Next=new int[Plen];//Next数组存储位置  
  
    GetNext(p);//求得Next数组  
  
    unsigned int i=0;//在S串中的下标  
    unsigned int j=0;//在P串中的下标  
    unsigned int count=0;//匹配串出现的次数
    while(i<Slen)  
    {  
        if(j==-1||s.at(i)==p.at(j))  
        {  
            i++;  
            j++;  
        }  
        else  if(j<Plen)
            j=Next[j];  
        if(j==Plen)
        {
            count++;
            j=Next[Plen];
        }
    }   
    return count;
}
 
int main()  
{  
    //freopen("in.txt", "r", stdin);
    string str1;  
    string str2;  
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>str2;
        cin>>str1; 
         
        int pos=KMPSearch(str1,str2);  
        cout<<pos<<endl; 
    
    }
   
    return 0;  
}  

WA:自己写的，本地样例全过,不知道错那

#include <iostream>
#include <cstring>
using namespace std;
char T[1000000+5];
char V[1000000+5];
 void getnext(char *T,int next[])     
 {
     int j=0;
     next[0]=0;
     for(int i=1;i<strlen(T);i++)
     {
         while(j>0 && T[i]!=T[j]){
             j=next[j-1];
     }
    if(T[i]==T[j]) j++;
    next[i]=j;    
 }
 for(int i=strlen(T)-1;i>0;i--){     
     next[i]=next[i-1];
 }
 next[0]=-1;
 
 }

int kmp(char *T,char *V){             
     int next[1000000+5];
    int flag=0;
     getnext(V,next);
     int i=0,j=0;
     while(i<strlen(T) && j< strlen(V)){
         if(T[i]==V[j] ){
             i++;
             j++;
         }
         else{
             
             j=next[j];
             if(j==-1){
                 i++;j++;
             }
         }
         
         if(j==strlen(V)-1 && V[j]==T[i]){
             flag++; 
             j=next[j];
         }
     }
     return flag;
 }

int main(int argc, char *argv[]) {
    int n;
    cin>>n;
    while(n--){
        scanf("%s%s",V,T);
        if(n==0)
        cout<<kmp(T,V)<<endl;
        else
        cout<<kmp(T,V);
    }
}