loj6031.「雅礼集训 2017 Day1」字符串

题意

注意到询问串的长度\(k\)是给定的，同时\(\sum\limits len_w=kq\leqslant10^5\)，我们发现\(k,q\)之间一个大了另一个必定会小，因此我们对\(k\)分类讨论：
首先肯定要对\(s\)建一个后缀自动机，对每个点维护\(endpos\)集合的大小。

\(k\geqslant\sqrt{n}\)

这时\(q\leqslant\sqrt{n}\)，我们直接在\(SAM\)暴力匹配\(w\)，找出所有\(w\)的前缀对应的节点，之后枚举询问，倍增跳到\(w[l_i...r_i]\)这个节点，将答案加上该节点的\(endpos\)集合大小即可。

\(k<\sqrt{n}\)

这时\(m\)组询问中的种类很少，我们可以枚举\(w\)的所有子串，算出它在\([l,r]\)这段询问区间中出现的次数，乘上它的答案加到答案中。

code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define re register
#define pii pair<int,int>
#define mkp make_pair
#define fir first
#define sec second
const int maxn=2e5+10;
const int maxt=410;
int n,m,Q,K,tot,cnt_edge,t;
int head[maxn],size[maxn],pos[maxn],posl[maxn];
int f[maxn][18];
char s[maxn];
inline int read()
{
	char c=getchar();re int res=0,f=1;
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')res=res*10+c-'0',c=getchar();
	return res*f;
}
struct edge{int to,nxt;}e[maxn];
inline void add_edge(int u,int v)
{
	e[++cnt_edge].nxt=head[u];
	head[u]=cnt_edge;
	e[cnt_edge].to=v;
}
struct Query{int l,r;}qr[maxn];
struct SAM
{
	int last,tot;
	int fa[maxn],len[maxn];
	int ch[maxn][26];
	SAM(){last=tot=1;}
	inline void add(int c)
	{
		re int now=++tot;len[now]=len[last]+1;
		re int p=last;last=now;
		while(p&&!ch[p][c])ch[p][c]=now,p=fa[p];
		if(!p){fa[now]=1;return;}
		re int q=ch[p][c];
		if(len[q]==len[p]+1)fa[now]=q;
		else
		{
			re int nowq=++tot;len[nowq]=len[p]+1;
			memcpy(ch[nowq],ch[q],sizeof(ch[q]));
			fa[nowq]=fa[q];fa[q]=fa[now]=nowq;
			while(p&&ch[p][c]==q)ch[p][c]=nowq,p=fa[p];
		}
	}
}sam;
void dfs(int x)
{
	for(re int i=1;i<=17;i++)f[x][i]=f[f[x][i-1]][i-1];
	for(re int i=head[x];i;i=e[i].nxt)
	{
		re int y=e[i].to;
		f[y][0]=x;dfs(y);
		size[x]+=size[y];
	}
}
inline void solve1()
{
	vector<int>a[maxt][maxt];
	for(int i=1;i<=m;i++)a[qr[i].l][qr[i].r].push_back(i);
	while(Q--)
	{
		scanf("%s",s+1);
		int l=read()+1,r=read()+1;
		ll res=0;
		for(int i=1;i<=K;i++)
			for(int j=i,now=1;j<=K;j++)
			{
				now=sam.ch[now][s[j]-'a'];
				if(!now)break;
				int ql=lower_bound(a[i][j].begin(),a[i][j].end(),l)-a[i][j].begin();
				int qr=upper_bound(a[i][j].begin(),a[i][j].end(),r)-a[i][j].begin();
				res+=1ll*(qr-ql)*size[now];
			}
		printf("%lld\n",res);
	}
}
inline int find(int l,int r)
{
	re int now=pos[r];
	for(re int i=17;~i;i--)if(sam.len[f[now][i]]>=min(r-l+1,posl[r]))now=f[now][i];
	return now;
}
inline int solve(int l,int r)
{
	if(!pos[r]||posl[r]<(r-l+1))return 0;;
	re int now=find(l,r);
	if(sam.len[now]>=r-l+1)return size[now];
	else return 0;
}
inline void solve2()
{
	while(Q--)
	{
		scanf("%s",s+1);
		re int l=read()+1,r=read()+1;
		re int now=1,nowl=0,res=0;
		for(re int i=1;i<=K;i++)
		{
			re int c=s[i]-'a';
			while(!sam.ch[now][c]&&now)now=sam.fa[now],nowl=sam.len[now];
			if(!now){now=1;pos[i]=1;nowl=0;posl[i]=0;continue;}
			now=sam.ch[now][c];pos[i]=now;
			nowl++;posl[i]=nowl;
		}	
		for(re int i=l;i<=r;i++)res+=solve(qr[i].l,qr[i].r);
		printf("%d\n",res);
	}
}
int main()
{
	//freopen("test.in","r",stdin);
	//freopen("test.out","w",stdout);
	n=read(),m=read(),Q=read(),K=read();t=330;
	scanf("%s",s+1);
	for(re int i=1;i<=n;i++)sam.add(s[i]-'a'),size[sam.last]=1;
	for(re int i=2;i<=sam.tot;i++)add_edge(sam.fa[i],i);
	dfs(1);
	for(re int i=1;i<=m;i++)qr[i].l=read()+1,qr[i].r=read()+1;
	if(K<=t)solve1();
	else solve2();
	return 0;
}