[51nod Round15 E ]Danganronpa

　　AC自动机,树上莫队，树状数组。。

　　比赛的时候完全看不出来...赛后去膜了一波网上题解才明白= =

　　离线，先把AC自动机建出来，把fail边反向建出fail树。

　　发射言弹，就是将言弹结束节点的fail子树内点权+1；

　　查询证言受的伤害，就是查询证言的每个前缀的结束节点的点权和。

　　前缀的结束节点的点权和，也就是AC自动机上，根到证言结束位置路径上的点权和。

　　每一个查询,就是查询在一段时间内，根到证言结束位置路径上的点权和。。。

　　所以就树上莫队一波了。。前两维时间，第三维是AC自动机上的位置。。再来个树状数组维护一波...

　　时间复杂度O(n^(3/5)logn)。。。。卡时过的。。。还要注意一下块的大小。。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=100233<<1;
struct zs{int too,pre;}e[maxn];int tt,last[maxn];
struct ask{int x,tim,timl,id;}b[maxn];
int B[maxn];
int ch[maxn][26],fail[maxn],dl[maxn],cnt;
int dfn[maxn],mp[maxn],TIME,L[maxn],R[maxn],TIM;
int pos[maxn],pos1[maxn],tim1[maxn],a[maxn];
int i,j,k,n,m;
ll sum,an[maxn];
bool u[maxn];

int ra;char rx;
inline int read(){
rx=getchar(),ra=0;
while(rx<'0'||rx>'9')rx=getchar();
while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra;
}

void dfs(int x){
L[x]=++TIM;
for(int i=last[x];i;i=e[i].pre)
dfs(e[i].too);
R[x]=TIM;//printf("    x:%d   l--r:%d %d\n",x,L[x],R[x]);
}
void dfs2(int x){
dfn[x]=++TIME,mp[TIME]=x;//printf("x:%d   dfn:%d\n",x,dfn[x]);
for(int i=0;i<26;i++)if(ch[x][i])
dfs2(ch[x][i]);
mp[++TIME]=x;
}
inline void insert(int a,int b){e[++tt].too=b,e[tt].pre=last[a],last[a]=tt;/*printf("%d-->%d\n",a,b);*/}

int t[maxn],t1[maxn];
inline void add1(int x){/*printf("  add1:%d\n",x);*/while(x<=cnt)t1[x]++,x+=x&-x;}
inline void del1(int x){/*printf("  del1:%d\n",x);*/while(x<=cnt)t1[x]--,x+=x&-x;}
inline int get1(int l,int r){
int sm=0;//printf("  get1:%d--%d",l,r);
l--;while(l)sm-=t1[l],l-=l&-l;
while(r)sm+=t1[r],r-=r&-r;/*printf(":   %d\n",sm);*/return sm;
}

inline int get(int x){
//    printf("get:%d",x);
int sm=0;while(x)sm+=t[x],x-=x&-x;//printf("  %d\n",sm);
return sm;
}
inline void add(int l,int r){
if(!l)return;//printf("add:   %d--%d\n",l,r);
sum+=get1(l,r);
while(l<=cnt)t[l]++,l+=l&-l;
r++;while(r<=cnt)t[r]--,r+=r&-r;
}
inline void del(int l,int r){
if(!l)return;//printf("del:  %d--%d\n",l,r);
sum-=get1(l,r);
while(l<=cnt)t[l]--,l+=l&-l;
r++;while(r<=cnt)t[r]++,r+=r&-r;
}

void getfail(){
int l=0,r=1,i,p,now,j;dl[1]=0;
while(l<r){
now=dl[++l];
for(i=0;i<26;i++)if(ch[now][i]){
dl[++r]=j=ch[now][i];
for(p=fail[now];p&&!ch[p][i];p=fail[p]);
fail[j]=!now?0:ch[p][i],insert(fail[j],j);
}
}
}

bool cmp(ask a,ask b){
return B[a.timl]==B[b.timl]?(B[a.tim]==B[b.tim]?a.x<b.x:B[a.tim]<B[b.tim]):B[a.timl]<B[b.timl];
}
int main(){
m=read();char id[13],c[2];int x,tm=0,numb=0,n=0,n1=0;
for(i=1;i<=m;i++){
scanf("%s",id);
if(id[0]=='I'){
n++,scanf("%s",c),c[0]-='a',x=read();
if(!ch[pos[x]][c[0]])ch[pos[x]][c[0]]=++cnt;//printf(" ..     %d %d\n",pos[x],c[0]);
pos
=ch[pos[x]][c[0]];//,tim
=tm;
}
if(id[0]=='A'){
n1++,scanf("%s",c),c[0]-='a',x=read();
if(!ch[pos1[x]][c[0]])ch[pos1[x]][c[0]]=++cnt;//,printf("      %d %d\n",pos1[x],c[0]);
pos1[n1]=ch[pos1[x]][c[0]],tim1[n1]=tm;
}
if(id[0]=='S')
tm++,a[tm]=pos[read()];
if(id[0]=='Q')
b[++numb].x=read(),b[numb].tim=tm,b[numb].timl=tim1[b[numb].x]+1,b[numb].id=numb;
}
cnt++;
getfail();
dfs(0),dfs2(0);int kuai=(int)pow(tm,1.8/3)+3;
for(i=0;i<=tm+1;i++)B[i]=(i+kuai-1)/kuai;
for(i=1;i<=numb;i++)b[i].x=dfn[pos1[b[i].x]];
sort(b+1,b+1+numb,cmp);

int l=1,r=0,pos=1;L[0]=R[0]=0,u[mp[1]]=1;
for(i=1;i<=numb;i++){
//        printf("ask:  %d--%d  %d--%d  x:%d   sum:%lld\n",b[i].timl,b[i].tim,B[b[i].timl],B[b[i].tim],b[i].x,sum);
while(l>b[i].timl)l--,add(L[a[l]],R[a[l]]);
while(r<b[i].tim)r++,add(L[a[r]],R[a[r]]);
while(l<b[i].timl)del(L[a[l]],R[a[l]]),l++;
while(r>b[i].tim)del(L[a[r]],R[a[r]]),r--;//printf("sum:  %lld\n",sum);
while(pos>b[i].x){
if(u[mp[pos]])sum-=get(L[mp[pos]]),del1(L[mp[pos]]);else sum+=get(L[mp[pos]]),add1(L[mp[pos]]);
u[mp[pos]]^=1,pos--;
}
while(pos<b[i].x){
pos++;//printf("x: %d   dfn:%d  L:%d\n",mp[pos],pos,L[mp[pos]]);
if(u[mp[pos]])sum-=get(L[mp[pos]]),del1(L[mp[pos]]);else sum+=get(L[mp[pos]]),add1(L[mp[pos]]);
u[mp[pos]]^=1;
}
an[b[i].id]=sum;
}
for(i=1;i<=numb;i++)printf("%lld\n",an[i]);
}

View Code