首页 > 代码库 > BZOJ 2434 阿狸的打字机
BZOJ 2434 阿狸的打字机
题解网上都有。。。要注意下细节。。。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<queue> #define maxn 100500 #define maxv 100500 #define maxe 200500 using namespace std; int ls,m,x,y,son[maxn][27],root,tot=0,cnt=0,t[maxn],fail[maxn],ans[maxn]; int l[maxv],r[maxv],pos[maxn],n=0,fath[maxv],g[maxv],nume=1,p=1; queue <int> qs; struct query { int x,y,id; query (int x,int y,int id):x(x),y(y),id(id) {} query () {} }q[maxn]; struct edge { int v,nxt; }e[maxe]; char s[maxn]; bool cmp(query x,query y) {return x.y<y.y;} int lowbit(int x) {return (x&(-x));} void addedge(int u,int v) { e[++nume].v=v; e[nume].nxt=g[u]; g[u]=nume; } void insert_AC() { int now=root; for (int i=0;i<ls;i++) { if (s[i]==‘B‘) now=fath[now]; else if (s[i]==‘P‘) pos[++n]=now; else { int nb=s[i]-‘a‘+1; if (!son[now][nb]) {son[now][nb]=++tot;fath[tot]=now;} now=son[now][nb]; } } } void build_AC() { qs.push(root); while (!qs.empty()) { int head=qs.front();qs.pop(); for (int i=1;i<=26;i++) { if (!son[head][i]) son[head][i]=son[fail[head]][i]; else { if (!head) fail[son[head][i]]=root; else fail[son[head][i]]=son[fail[head]][i]; qs.push(son[head][i]); } } } for (int i=1;i<=tot;i++) { addedge(fail[i],i); addedge(i,fail[i]); } } void dfs(int now,int fath) { l[now]=r[now]=++cnt; for (int i=g[now];i;i=e[i].nxt) { int v=e[i].v; if (v==fath) continue; dfs(v,now);r[now]=max(r[now],r[v]); } } void add_query() { scanf("%d",&m); for (int i=1;i<=m;i++) { scanf("%d%d",&x,&y); q[i]=query(x,y,i); } sort(q+1,q+m+1,cmp); } void add(int x,int val) { for (int i=x;i<=cnt;i+=lowbit(i)) t[i]+=val; } int ask(int x) { int ret=0; for (int i=x;i>=1;i-=lowbit(i)) ret+=t[i]; return ret; } void ask_() { int now=root,kr=0; for (int i=0;i<ls;i++) { if (s[i]==‘B‘) {add(l[now],-1);now=fath[now];} else if (s[i]==‘P‘) { kr++; for (;p<=m;p++) { if (q[p].y!=kr) break; ans[q[p].id]=ask(r[pos[q[p].x]])-ask(l[pos[q[p].x]]-1); } } else { now=son[now][s[i]-‘a‘+1]; add(l[now],1); } } } int main() { n=root=tot=0; scanf("%s",s);ls=strlen(s); insert_AC();build_AC(); dfs(root,root);add_query(); ask_(); for (int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }
BZOJ 2434 阿狸的打字机
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。