Searching the String ZOJ - 3228 AC自动机查询升级版

题意：
先给你一个不超过1000000长度的大串s；接下来输入一个n代表接下来输入的小串个数，小串长度不超过6。
小串分两种类型0和1类型。
0类型表示小串在大串中的最大匹配个数就是常规的AC自动机的做法。
1类型表示小串在大串中不能重合的最大匹配数。
依次输出结果.（所有的串只包含小写字母）
按样例输出，注意每组测试数据后有一个换行。

题意我不想写了抄的，抄这里的 （不好意思啦）

0 类型的就是最开始的模板题

1 类型的处理方式就是，在建立字典树的时候弄一个dep数组，记录每一个节点的深度

然后在query的过程中查询上一个单词最后一个字母出现的位置。

if (i - last[temp] >= dep[temp]) ans[temp][1]++, last[temp] = i;

这样就可以了。

 #include <set>
 #include <map>
 #include <stack>
 #include <queue>
 #include <cmath>
 #include <ctime>
 #include <cstdio>
 #include <string>
 #include <vector>
 #include <cstring>
 #include <iostream>
 #include <algorithm>

 #define  pi    acos(-1.0)
 #define  eps   1e-9
 #define  fi    first
 #define  se    second
 #define  rtl   rt<<1
 #define  rtr   rt<<1|1
 #define  bug                printf("******\n")
 #define  mem(a, b)          memset(a,b,sizeof(a))
 #define  name2str(x)        #x
 #define  fuck(x)            cout<<#x" = "<<x<<endl
 #define  sfi(a)             scanf("%d", &a)
 #define  sffi(a, b)         scanf("%d %d", &a, &b)
 #define  sfffi(a, b, c)     scanf("%d %d %d", &a, &b, &c)
 #define  sffffi(a, b, c, d) scanf("%d %d %d %d", &a, &b, &c, &d)
 #define  sfL(a)             scanf("%lld", &a)
 #define  sffL(a, b)         scanf("%lld %lld", &a, &b)
 #define  sfffL(a, b, c)     scanf("%lld %lld %lld", &a, &b, &c)
 #define  sffffL(a, b, c, d) scanf("%lld %lld %lld %lld", &a, &b, &c, &d)
 #define  sfs(a)             scanf("%s", a)
 #define  sffs(a, b)         scanf("%s %s", a, b)
 #define  sfffs(a, b, c)     scanf("%s %s %s", a, b, c)
 #define  sffffs(a, b, c, d) scanf("%s %s %s %s", a, b,c, d)
 #define  FIN                freopen("../date.txt","r",stdin)
 #define  gcd(a, b)          __gcd(a,b)
 #define  lowbit(x)          x&-x
 #define  IO                 iOS::sync_with_stdio(false)

 using namespace std;
 typedef long long LL;
 typedef unsigned long long ULL;
 const ULL seed = ;
 const LL INFLL = 0x3f3f3f3f3f3f3f3fLL;
 const int maxn = 1e5 + ;
 const int maxm = 8e6 + ;
 const int INF = 0x3f3f3f3f;
 const int mod = 1e9 + ;

 int n, vis[maxn], last[*maxn], ans[*maxn][], pos[maxn], dep[*maxn];
 char buf[maxn], str[maxn][];

 struct Aho_Corasick {
     int next[][], fail[], End[];
     int root, cnt;

     int newnode() {
         for (int i = ; i < ; i++) next[cnt][i] = -;
         End[cnt++] = ;
         return cnt - ;
     }

     void init() {
         cnt = ;
         root = newnode();
         dep[root]=;
     }

     int insert(char buf[]) {
         int len = strlen(buf);
         int now = root;
         for (int i = ; i < len; i++) {
             if (next[now][buf[i] - 'a'] == -) {
                 next[now][buf[i] - 'a'] = newnode();
                 dep[next[now][buf[i] - 'a']] = i + ;
             }
             now = next[now][buf[i] - 'a'];
         }
         End[now] = ;
         return now;
     }

     void build() {
         queue<int> Q;
         fail[root] = root;
         for (int i = ; i < ; i++)
             if (next[root][i] == -) next[root][i] = root;
             else {
                 fail[next[root][i]] = root;
                 Q.push(next[root][i]);
             }
         while (!Q.empty()) {
             int now = Q.front();
             Q.pop();
             for (int i = ; i < ; i++)
                 if (next[now][i] == -) next[now][i] = next[fail[now]][i];
                 else {
                     fail[next[now][i]] = next[fail[now]][i];
                     Q.push(next[now][i]);
                 }
         }
     }

     void query(char buf[]) {
         int len = strlen(buf);
         int now = root;
         mem(ans,);
         mem(last, -);
         for (int i = ; i < len; i++) {
             now = next[now][buf[i] - 'a'];
             int temp = now;
             while (temp != root) {
                 ans[temp][]++;
                 if (i - last[temp] >= dep[temp]) ans[temp][]++, last[temp] = i;
                 temp = fail[temp];
             }
         }
     }

     void debug() {
         for (int i = ; i < cnt; i++) {
             printf("id = %3d,fail = %3d,end = %3d,chi = [", i, fail[i], End[i]);
             for (int j = ; j < ; j++) printf("%2d", next[i][j]);
             printf("]\n");
         }
     }
 } ac;

 int main() {
   //  FIN;
     int cas = ;
     while (~sfs(buf)) {
         ac.init();
         sfi(n);
         for (int i = ; i < n; ++i) {
             scanf("%d%s", &vis[i], str[i]);
             pos[i] = ac.insert(str[i]);
         }
         ac.build();
         ac.query(buf);
         printf("Case %d\n", cas++);
         for (int i = ; i < n; ++i) printf("%d\n", ans[pos[i]][vis[i]]);
         printf("\n");
     }
     return ;
 }