好题啊……orz PoPoQQQ 大爷

  一道相似的题目:【BZOJ】【3172】【TJOI2013】单词

  那道题也是在fail树上数有多少个点,只不过这题是在x的fail树上数有多少个y的点

  感觉好难搞啊……那么我们不妨反过来……离线做?

 

  既然是fail树!那么就看可以有dfs序,那么我们可以先找到整棵树上所有y的点,再看有哪些是在x的fail树上的,怎么做?x的fail树,对应的是整个dfs序上的一个区间!其实就是令dfs序上字符串y的点为1,其他为0,求一个区间和!用BIT就可以搞啦~

 

  算法的大体框架就是:建AC自动机,搞出fail树的dfs序,在Trie树上dfs(枚举串y),将当前经过的这条链上的所有点在dfs序中对应的位置 置为1,如果走到某个字符串的结束点,一并处理所有与这个串相关的询问(对一个序列,分别查询多个区间和)。

 

TLE:一开始依旧写的以前的AC模板……其实那个是Trie图的,因为要出边补全,所以无论是否有这条出边都要找一遍fail,而这题只是一个AC自动机,不需要出边补全,所以可以大量减少找fail的复杂度……就轻松过了……

/**************************************************************
    Problem: 2434
    User: Tunix
    Language: C++
    Result: Accepted
    Time:440 ms
    Memory:18716 kb
****************************************************************/
 
//BZOJ 2434
#include<vector>
#include<string>
#include<map>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define rep(i,n) for(int i=0;i<n;++i)
#define F(i,j,n) for(int i=j;i<=n;++i)
#define D(i,j,n) for(int i=j;i>=n;--i)
#define pb push_back
using namespace std;
inline int getint(){
    int v=0,sign=1; char ch=getchar();
    while(ch<'0'||ch>'9'){ if (ch=='-') sign=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){ v=v*10+ch-'0'; ch=getchar();}
    return v*sign;
}
const int N=1e5+10,INF=~0u>>2;
typedef long long LL;
/******************tamplate*********************/
 
int to[N],nxt[N],head[N],cnt;
void add(int x,int y){
    to[++cnt]=y; nxt[cnt]=head[x]; head[x]=cnt;
}
 
 
int n,m,tot=1,num,pos[N];
struct node{
    int ch[26],fail,sign,fa;
}t[N];
char s1[N];
inline int id(char ch){return ch-'a';}
 
queue<int>Q;
void make_fail(){
    Q.push(1);
    while(!Q.empty()){
        int x=Q.front(),j,y; Q.pop();
        add(t[x].fail,x);
        rep(i,26){
            if (!t[x].ch[i]) continue;
            j=t[x].fail;
            while(j && !t[j].ch[i]) j=t[j].fail;
            y=t[x].ch[i];
            t[y].fail=j ? t[j].ch[i] : 1;
            Q.push(y);
        }
    }
}
 
struct ques{int x,num;};
vector<ques>G[N];
int st[N],ed[N],c[N],dfs_clock,ans[N];
void ad(int x,int v){
    for(int i=x;i<=tot;i+=i&(-i)) c[i]+=v;
}
int sum(int x){
    if (x==0) return 0;
    int r=0;
    for(int i=x;i;i-=i&(-i)) r+=c[i];
    return r;
}
void dfs(int x){
    st[x]=++dfs_clock;
    for(int i=head[x];i;i=nxt[i])
        dfs(to[i]);
    ed[x]=dfs_clock;
}
 
int main(){
#ifndef ONLINE_JUDGE
    freopen("2434.in","r",stdin);
    freopen("2434.out","w",stdout);
#endif
    scanf("%s",s1);
    int l=strlen(s1);
    //build trie
    int x=1;
    rep(i,l){
        if (s1[i]=='P'){
            t[x].sign=++num;
            pos[num]=x;
        }
        else if (s1[i]=='B') x=t[x].fa;
        else{
            int y=id(s1[i]);
            if (!t[x].ch[y]) t[x].ch[y]=++tot,t[tot].fa=x;
            x=t[x].ch[y];
        }
    }
    make_fail();
    //end build
    int T=getint();
    F(i,1,T){
        int x=getint(),y=getint();
        G[y].pb((ques){x,i});
    }
    dfs(1);
    x=1;
    rep(i,l){
        if (s1[i]=='P'){
            int y=t[x].sign;
            rep(j,G[y].size()){
                int l=st[pos[G[y][j].x]],r=ed[pos[G[y][j].x]];
                ans[G[y][j].num]=sum(r)-sum(l-1);
            }
        }else if (s1[i]=='B'){
            ad(st[x],-1);
            x=t[x].fa;
        }else{
            x=t[x].ch[id(s1[i])];
            ad(st[x],1);
        }
    }
    F(i,1,T) printf("%d\n",ans[i]);
    return 0;
}
View Code

相关文章:

  • 2021-06-25
  • 2022-02-15
  • 2022-03-05
  • 2021-05-30
  • 2021-05-20
  • 2021-07-24
  • 2021-06-11
  • 2022-12-23
猜你喜欢
  • 2021-09-02
  • 2021-10-01
  • 2021-07-16
  • 2021-12-22
  • 2021-09-10
  • 2021-12-28
  • 2022-01-16
相关资源
相似解决方案