HDU-3695 Computer Virus on Planet Pandora

HDU-3695 Computer Virus on Planet Pandora

题意：电脑中病毒了， 现在n钟病毒指令， 然后有一个电脑指令， 看一下这个电脑指令中了几个病毒， 如果电脑种了某一个病毒， 那么就有子串是病毒指令， 或者 子串的反串是病毒指令， 现在问电脑指令一共感染了多少病毒。

题解：AC自动机，然后正向匹配一遍， 反向匹配一遍。

代码：

 #include<bits/stdc++.h>
 using namespace std;
 #define LL long long
 #define ULL unsigned LL
 #define fi first
 #define se second
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1
 #define max3(a,b,c) max(a,max(b,c))
 #define min3(a,b,c) min(a,min(b,c))
 const int INF = 0x3f3f3f3f;
 const LL mod = 1e9+;
 typedef pair<int,int> pll;
 const int N = 1e6+;
 int tot = ;
 char str[N*], tmp[N*];
 int trie[N*][], cnt[N*], fair[N*];
 void init(){
     for(int i = ; i < tot; i++){
         for(int j = ; j < ; j++)
             trie[i][j] = ;
     }
     tot = ;
 }
 void Insert(){
     int rt = , len = strlen(str);
     for(int i = ; i < len; i++){
         int id = str[i] - 'A';
         if(!trie[rt][id]) {
             cnt[tot] = ;
             fair[tot] = ;
             trie[rt][id] = tot++;
         }
         rt = trie[rt][id];
     }
     cnt[rt]++;
 }
 void Build_tree(){
     queue<int> q;
     q.push();
     int p;
     while(!q.empty()){
         int tmp = q.front();
         q.pop();
         for(int j = ; j < ; j++){
             if(trie[tmp][j]){
                 if(tmp == )
                     fair[trie[tmp][j]] = ;
                 else {
                     p = fair[tmp];
                     while(p){
                         if(trie[p][j]){
                             fair[trie[tmp][j]] = trie[p][j];
                             break;
                         }
                         else p = fair[p];
                     }
                     if(!p) fair[trie[tmp][j]] = ;
                 }
                 q.push(trie[tmp][j]);
             }
         }
     }
 }
 int Find(int len){
     int ret = , rt = ;
     for(int i = ; i < len; i++){
         int id = str[i] - 'A';
         while(rt !=   && !trie[rt][id]) rt = fair[rt];;
         if(trie[rt][id]) rt = trie[rt][id];
         int p = rt;
         while(p != ){
             if(cnt[p] >= ){
                 ret += cnt[p];
                 cnt[p] = -;
             }
             else break;
             p = fair[p];
         }
     }
     rt = ;
     for(int i = len - ; i >= ; i--){
         int id = str[i] - 'A';
         while(rt !=   && !trie[rt][id]) rt = fair[rt];
         if(trie[rt][id]) rt = trie[rt][id];
         int p = rt;
         while(p != ){
             if(cnt[p] >= ){
                 ret += cnt[p];
                 cnt[p] = -;
             }
             else break;
             p = fair[p];
         }
     }
     return ret;
 }
 int main(){
     int t;
     scanf("%d", &t);
     while(t--){
         int n;
         init();
         scanf("%d", &n);
         while(n--){
             scanf("%s", str);
             Insert();
         }
         Build_tree();
         scanf("%s", tmp);
         int len = strlen(tmp);
         int ccc = ;
         for(int i = ; i < len; i++){
             if(tmp[i] != '['){
                 str[ccc++] = tmp[i];
             }
             else {
                 int _c = ;
                 i++;
                 while(isdigit(tmp[i]))
                     _c = _c * + (int)(tmp[i]-''), i++;
                 while(_c--)
                     str[ccc++] = tmp[i];
                 i++;
             }
         }
         printf("%d\n", Find(ccc));
     }
     return ;
 }