[BZOJ3881][Coci2015]Divljak（AC自动机+fail树+lca+树状数组）

题目：

我是超链接

题解：

实际上的问题就是给出了一些短串，和一些长串，问每个短串出现在了多少个长串中。
我们将短串加入trie，然后建立AC自动机，构建出trie树。

先考虑一个比较暴力的思路，假设长串中的某个位置匹配到了AC自动机中的x节点，那么x节点在fail树上到根路径上的所有短串的结尾节点的end都要+1。我们的题就fail了，这样好慢啊

以上这种区间修改单点查询的问题可以转化成单点打标记然后查询子树和。
所以我将所有能匹配到的节点记录下来，因为求短串的end节点是取并集，那么就需要按照dfs序排序然后将相邻两个点的lca-1，做一个差分，用树状数组维护就挺好的。

对拍n久查出来的错误：记得开f和mi数组的时候要比sz大一些。。。

代码：

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=2000005;
const int sz=25;
char st[N];int l,cnt,ch[N][30],nc[N],fail[N],ans[N],c[N],mi[sz+5],in[N],num,h[N],out[N],is_end[N];
int tot,nxt[N],point[N],v[N],f[N][sz+5];
void addline(int x,int y){++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;}
void insert(int id)
{
    scanf("%s",st);
    int now=0;l=strlen(st);
    for (int i=0;i<l;i++)
    {
        int x=st[i]-'a';
        if (!ch[now][x]) ch[now][x]=++cnt;
        now=ch[now][x];
    }
    is_end[id]=now;
}
void sp()
{
    queue<int>q;
    for (int i=0;i<26;i++)
      if (ch[0][i]) q.push(ch[0][i]);
    while (!q.empty())
    {
        int now=q.front(); q.pop();
        for (int i=0;i<26;i++)
        {
            if (!ch[now][i])
            {
                ch[now][i]=ch[fail[now]][i];
                continue;
            }
            fail[ch[now][i]]=ch[fail[now]][i];
            q.push(ch[now][i]);
        }
    }
}
void dfs(int x)
{
    in[x]=++num;
    for (int i=1;i<sz;i++)
      if (h[x]<mi[i]) break;
      else f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      {
        h[v[i]]=h[x]+1;
        f[v[i]][0]=x;
        dfs(v[i]);
      }
    out[x]=num;
}
int lca(int x,int y)
{
    if (h[x]<h[y]) swap(x,y);
    int k=h[x]-h[y];
    for (int i=0;i<sz;i++)
      if (k>>i&1) x=f[x][i];
    if (x==y) return x;
    for (int i=sz-1;i>=0;i--)
      if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void add(int loc,int v){for (int i=loc;i<=num;i+=i&(-i)) c[i]+=v;}
int qurry(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i)) ans+=c[i];
    return ans;
}
int cmp(int a,int b){return in[a]<in[b];}
void ac()
{
    scanf("%s",st);
    int now=0,lj=0;l=strlen(st);
    for (int i=0;i<l;i++)
    {
        while (now && !ch[now][st[i]-'a']) now=fail[now];
        if (ch[now][st[i]-'a']) now=ch[now][st[i]-'a'];
        nc[++lj]=now;
    }
    sort(nc+1,nc+lj+1,cmp);
    add(in[nc[1]],1);
    for (int i=2;i<=lj;i++)
    {
        add(in[nc[i]],1);
        add(in[lca(nc[i-1],nc[i])],-1);
    } 
}
int main()
{
    mi[0]=1;for (int i=1;i<sz;i++) mi[i]=mi[i-1]*2;
    int n,q;scanf("%d",&n);
    for (int i=1;i<=n;i++) insert(i);
    sp();
    for (int i=1;i<=cnt;i++) addline(fail[i],i);
    h[0]=1;dfs(0);
    scanf("%d",&q);
    while (q--)
    {
        int id;scanf("%d",&id);
        if (id==1) ac();
        else 
        {
            int x;scanf("%d",&x);x=is_end[x];
            printf("%d\n",qurry(out[x])-qurry(in[x]-1));
        }
    }
}