Pty loves string
建立Border树后,发现可以转化成两个子树中相同点的数量,时间戳转化为连续的区间后相当于有两个数组,每次给两个区间,问区间相同点权的数目。
第一个数组作为区间,第二个数组作为权值。将第一个数组建立主席树,每次插入的是第一个区间位置的权值映射到第二个数组中的权值。然后就是区间查询。
#include<bits/stdc++.h>using namespace std;
const int N=200010;
char s[N];
int n,m;
int ne1[N],ne2[N];
vector<int> e1[N],e2[N];
int dfn1[N],dfn2[N],sz1[N],sz2[N],timestamp;
int id[N];
struct node
{int l,r,v;
}tree[N*40];
int rt[N],cnt;
void insert(int &u,int o,int l,int r,int v)
{tree[u=++cnt]=tree[o];tree[u].v++;if(l==r) return;int mid=l+r>>1;if(v<=mid)insert(tree[u].l,tree[o].l,l,mid,v);elseinsert(tree[u].r,tree[o].r,mid+1,r,v);
}
int query(int u,int o,int l,int r,int L,int R)
{if(L<=l&&r<=R) return tree[u].v-tree[o].v;int v=0;int mid=l+r>>1;if(L<=mid) v+=query(tree[u].l,tree[o].l,l,mid,L,R);if(R> mid) v+=query(tree[u].r,tree[o].r,mid+1,r,L,R);return v;
}
void dfs1(int u)
{sz1[u]=1;dfn1[u]=++timestamp;id[timestamp]=u;for(int v:e1[u]){dfs1(v);sz1[u]+=sz1[v];}
}
void dfs2(int u)
{sz2[u]=1;dfn2[u]=++timestamp;for(int v:e2[u]){dfs2(v);sz2[u]+=sz2[v];}
}
void init()
{for(int i=1;i<=cnt;i++) tree[i]={0,0,0};cnt=0;for(int i=0;i<=n+1;i++) e1[i].clear(),e2[i].clear();ne1[1]=0;for(int i=2,j=0;i<=n;i++){while(j&&s[i]!=s[j+1]) j=ne1[j];if(s[i]==s[j+1]) j++;ne1[i]=j;}ne2[n]=n+1;for(int i=n-1,j=n+1;i;i--){while(j<=n&&s[i]!=s[j-1]) j=ne2[j];if(s[i]==s[j-1]) j--;ne2[i]=j;}
}
void build()
{for(int i=1;i<=n;i++) e1[ne1[i]].push_back(i),e2[ne2[i]].push_back(i);timestamp=0;dfs1(0);timestamp=0;dfs2(n+1);for(int i=1;i<=n+1;i++) insert(rt[i],rt[i-1],1,n+1,dfn2[id[i]+1]);
}
int main()
{int Tc;scanf("%d",&Tc);while(Tc--){scanf("%d%d%s",&n,&m,s+1);init();build();while(m--){int x,y;scanf("%d%d",&x,&y);if(x+y>n) {puts("0");continue;}y=n-y+1;printf("%d\n",query(rt[dfn1[x]+sz1[x]-1],rt[dfn1[x]-1],1,n+1,dfn2[y],dfn2[y]+sz2[y]-1));}}return 0;
}