【数据结构与算法】[hihoCode] #1015 : KMP算法（解决Time Limit Exceeded）

博主在做hihoCode的#1015 : KMP算法时，用了传统的KMP算法，但是一直报错“Time Limit Exceeded ”，网上查询很久，不断修改代码，最终Accepted。所以记录其中最关键部分，以备大家交流~

---

题设求，匹配串中有多少个模式串？

传统KMP是返回模式串在匹配串中的位置，下面进入正题，怎样利用KMP算法的思想（匹配串永不回溯）？正解是求next数组时，增加一位。如图所示：

为什么要增加一位next？

由于本题求的是匹配串中有多少个模式串，所以当模式串遍历一遍后，需要继续遍历匹配串直至匹配串也遍历一遍，才能统计出有多少个模式串。找到一个模式串，再找第二个时，令j=next[j]，就能使匹配串不回溯，从而大大节省运行时间，形象图解如下：

代码分享

import java.util.*;

public class Main{
    public static int[] next(String pattern){
        int[] next=new int[pattern.length()+1];
        int k=-1;
        int j=0;
        next[0]=k;
        while(j<pattern.length()){
            if (k==-1||pattern.charAt(k)==pattern.charAt(j)){
                next[++j]=++k;
            }else{
                k=next[k];
            }
        }
        return next;
    }
    public static int search(String match,String pattern,int[] next){
        int i=0;
        int j=0;
        int count=0;
        while(i<match.length()){
            if(j==-1||pattern.charAt(j)==match.charAt(i)){
                j++;
                i++;
            }else{
                j=next[j];
            }
            if(j==pattern.length()){

                count++;

                j=next[j];
            }
        }

        return count;
    }
    public static void main(String[] args){
        Scanner sc=new Scanner(System.in);
        int num=Integer.valueOf(sc.nextLine());
        for(int i=0;i<num;i++){
            String pattern=sc.nextLine();
            String match=sc.nextLine();
            int[] next=next(pattern);
            //System.out.println(Arrays.toString(next));
            int count=search(match,pattern,next);

            System.out.println(count);
        }

    }
}