java版AC自动机

 1 class Trie
 2 {
 3     int [][]Next=new int[500005][128];
 4     int []fail=new int[500005];
 5     int []end=new int[500005];
 6     int root, L;
 7     int newnode()
 8     {
 9         for(int i=0;i<128;i++)
10             Next[L][i]=-1;
11         end[L++]=0;
12         return L-1;
13     }
14     void init()
15     {
16         L=0;
17         root=newnode();
18     }
19     void insert(byte buf[], int id)
20     {
21         int now=root;
22         for(int i=0; i<buf.length; i++)
23         {
24             if(Next[now][buf[i]]==-1)
25                 Next[now][buf[i]]=newnode();
26             now=Next[now][buf[i]];
27         }
28         end[now]=id;
29     }
30     void build()
31     {
32         Queue<Integer> q=new LinkedList<Integer>();
33         fail[root]=root;
34         for(int i=0; i<128; i++)
35         {
36             if(Next[root][i]==-1)
37                 Next[root][i]=root;
38             else
39             {
40                 fail[Next[root][i]]=root;
41                 q.add(Next[root][i]);
42             }
43         }
44         while(!q.isEmpty())
45         {
46             int now=q.poll();
47             for(int i=0; i<128; i++)
48             {
49                 if(Next[now][i]==-1)
50                     Next[now][i]=Next[fail[now]][i];
51                 else
52                 {
53                     fail[Next[now][i]]=Next[fail[now]][i];
54                     q.add(Next[now][i]);
55                 }
56             }
57         }
58     }
59     int query(byte buf[], int n, String s[])
60     {
61         int now=root;
62         int ans=0;
63         for(int i=0; i<buf.length; i++)
64         {
65             now=Next[now][buf[i]];
66             int temp=now;
67             while(temp!=root)
68             {
69                 ans+=end[temp];
70                 end[temp]=0;    
71                 temp=fail[temp];
72             }
73         }
74         return ans;
75     }
76 }

View Code

 

转载于:https://www.cnblogs.com/Empress/p/4632083.html