2434: [Noi2011]阿狸的打字机
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 3317 Solved: 1823
[ Submit][ Status][ Discuss]
Description
阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机。打字机上只有28个按键,分别印有26个小写英文字母和'B'、'P'两个字母。
经阿狸研究发现,这个打字机是这样工作的:
l 输入小写字母,打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。
l 按一下印有'B'的按键,打字机凹槽中最后一个字母会消失。
l 按一下印有'P'的按键,打字机会在纸上打印出凹槽中现有的所有字母并换行,但凹槽中的字母不会消失。
例如,阿狸输入aPaPBbP,纸上被打印的字符如下:
a
aa
ab
我们把纸上打印出来的字符串从1开始顺序编号,一直到n。打字机有一个非常有趣的功能,在打字机中暗藏一个带数字的小键盘,在小键盘上输入两个数(x,y)(其中1≤x,y≤n),打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。
阿狸发现了这个功能以后很兴奋,他想写个程序完成同样的功能,你能帮助他么?
Input
输入的第一行包含一个字符串,按阿狸的输入顺序给出所有阿狸输入的字符。
第二行包含一个整数m,表示询问个数。
接下来m行描述所有由小键盘输入的询问。其中第i行包含两个整数x, y,表示第i个询问为(x, y)。
Output
输出m行,其中第i行包含一个整数,表示第i个询问的答案。
Sample Input
3
1 2
1 3
2 3
Sample Output
1
0
HINT
1<=N<=10^5
Source
相信自己的想法,自己的程序,只要算法正确,一定能A的,不要轻易放弃.
题解
dfs o(n),查询(logn),总时间复杂度(nlogn).
闲话
一开始插入用栈存,每次从根开始插入新串结果超时,多次崩溃后仔仔细细·检查程序时间复杂度没有问题,只有插入不能保证,所以改一下插入方式就A了...要坚信自己的程序.
#include<stdio.h>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=400005;
char s[maxn],ss[maxn];
int top,cnt,idc,num1,num2,tot,in[maxn],out[maxn],mp[maxn],query[maxn],h[maxn],hh[maxn],c[maxn],T,fa[maxn];
queue<int> q;
vector<pair<int,int> > G[maxn];
struct point{int c[26],fail;}mm[maxn];
inline const int read(){
register int f=1,x=0;
register char ch=getchar();
while(ch<'0'||ch>'9') ch=getchar();
while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
return f*x;
}
struct edge{
int nxt,v;
}e[maxn],ee[maxn];
inline void adde(int u,int v){ee[++num1].v=v,ee[num1].nxt=hh[u],hh[u]=num1;}
inline void add(int u,int v){e[++num2].v=v,e[num2].nxt=h[u],h[u]=num2;}
inline void build(){
int p=0;
for(int i=0;ss[i];i++){
if(ss[i]=='P') {mp[++cnt]=p;continue;}
if(ss[i]=='B') {p=fa[p];continue;}
int index=ss[i]-'a';
if(!mm[p].c[index]) mm[p].c[index]=++tot,fa[tot]=p,add(p,tot);
p=mm[p].c[index];
}
}
inline void bfs(){
for(int i=0;i<26;i++){
int v=mm[0].c[i];
if(v) {mm[v].fail=0;q.push(v);}
}
while(!q.empty()){
int u=q.front();q.pop();
for(int i=0;i<26;i++){
int v=mm[u].c[i];
if(!v){mm[u].c[i]=mm[mm[u].fail].c[i];continue;}
mm[v].fail=mm[mm[u].fail].c[i];
q.push(v);
}
}
}
void dfs(int u){
in[u]=++idc;
for(int i=hh[u];i;i=ee[i].nxt) dfs(ee[i].v);
out[u]=idc;
}
inline void modify(int x,int delta){
if(!x) return;
for(int i=x;i<=idc;i+=i&-i) c[i]+=delta;
}
inline int qquery(int x){
if(!x) return 0;
int tmp=0;
for(int i=x;i>0;i-=i&-i) tmp+=c[i];
return tmp;
}
void dfs1(int u){
modify(in[u],1);int siz=G[u].size();
if(siz) for(int i=0;i<siz;i++) query[G[u][i].second]=qquery(out[G[u][i].first])-qquery(in[G[u][i].first]-1);
for(int i=h[u];i;i=e[i].nxt){
int v=e[i].v;
dfs1(v);
}
modify(in[u],-1);
}
int main(){
scanf("%s",ss);
build();
bfs();
for(int i=1;i<=tot;i++) adde(mm[i].fail,i);
dfs(0);
T=read();
for(int i=1;i<=T;i++){
int x=read(),y=read();
G[mp[y]].push_back(make_pair(mp[x],i));
}
dfs1(0);
for(int i=1;i<=T;i++) printf("%d\n",query[i]);
}