[HDU 4787] GRE Words Revenge (AC自动机)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4787

题目大意：

给你若干个单词，查询一篇文章里出现的单词数。。

就是被我水过去的。。。暴力重建AC自动机- -然后暴力查找。。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <queue>
 #include <map>
 #include <string>
 #include <set>
 using namespace std;

 #define CHARSET 2
 #define MAX_NODE 1010000

 int T,N;
 char inp[];

 struct trieNode{
     int ch[CHARSET];
     int cnt;
     //bool vis;
 };
 trieNode trie[MAX_NODE];
 int ptr,fail[MAX_NODE];

 void init(){
     memset(fail,,sizeof(fail));
     ptr = ;
     for(int i=;i<CHARSET;i++){
         trie[].ch[i] = -;
     }
     trie[].cnt = ;
     //trie[0].vis = false;
 }

 int newNode(){
     ptr++;
     for(int i=;i<CHARSET;i++){
         trie[ptr].ch[i] = -;
     }
     trie[ptr].cnt = ;
     //trie[ptr].vis = false;
     return ptr;
 }

 void insert(const char* str,int L){
 //    printf("insert:");
     int len = strlen(str);
     int rt = ;
     L %= len;
     for(int i=;i<len;i++){
         //trie[rt].vis = true;
         int id = str[(i+L)%len] -'';
 //        printf("%d",id);
         if(trie[rt].ch[id]==-){
             trie[rt].ch[id] = newNode();
         }
         rt = trie[rt].ch[id];
     }
     //trie[rt].vis = true;
     trie[rt].cnt = ;
 //    puts("");
 }

 void build(){
     queue<int> q;
     //trie[0].vis = false;
     q.push();
     fail[] = -;
     while(!q.empty()){
         int t = q.front(); q.pop();
         for(int i=;i<CHARSET;i++){
             int v = trie[t].ch[i];
             if( v==- ) continue;
 //            if( trie[v].vis ){
 //                trie[v].vis = false;
 //            } else continue;
             int f = fail[t];
             while(f!=-&&trie[f].ch[i]==-){
                 f = fail[f];
             }
             if(f==-) fail[v] = ;
             else fail[v] = trie[f].ch[i];
             q.push(v);
         }
     }
 }

 int query(const char *str,int L){
 //    printf("query:");
     int res = ;
     int rt = ;
     int len = strlen(str);
     L %= len;
     for(int i=;i<len;i++){
         int id = str[(i+L)%len] - '';
 //        printf("%d",id);
         while(rt&&trie[rt].ch[id]==-) rt = fail[rt];
         rt = trie[rt].ch[id];

         if(rt==-){
             rt = ;
             continue;
         }

         int now = rt;
         while(now!=-){
             res += trie[now].cnt;
             now = fail[now];
         }
     }
 //    puts("");
     return res;
 }

 int main(){
 //    freopen("output.txt","w",stdout);
     scanf("%d",&T);
     for(int t=;t<=T;t++){
         int L = ;
         init();
         printf("Case #%d:\n",t);
         scanf("%d",&N);
         while(N--){
             scanf("%s",inp);
             if( inp[]=='+' ){
                 insert(inp+,L);
             } else {
                 build();
                 int ans = query(inp+,L);
                 L = ans;
                 printf("%d\n",ans);
             }
         }
     }
     return ;
 }