【Luogu3732】[HAOI2017]供给侧改革（Trie树）

【Luogu3732】[HAOI2017]供给侧改革（Trie树）

题面

洛谷

给定一个纯随机的\(01\)串，每次询问\([L,R]\)之间所有后缀两两之间的\(LCP\)的最大值。

题解

一个暴力的想法是构建\(SA\)之后把所有位置按照\(rank\)排序，每次询问相邻的两个\(LCP\)就行了，然后拿\(set\)维护插入的操作。

然而并没有用到串随机的性质。

既然是随机的，那么大力猜一猜他们的\(lcp\)的长度不会很长，大概估计一个\(40\)左右吧。

把询问离线挂在右端点上，从左往右把每个后缀的前\(40\)个位置插入\(trie\)数，记录可以取到\(LCP\)为每个值的时候的最靠右的端点，然后统计一下就好了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define MAX 100100
#define mp make_pair
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int n,Q,p[MAX],ans[MAX];char ch[MAX];
int L[MAX],R[MAX];
vector<int> A[MAX];
int fr[MAX];
struct Node{int ch[2],lst;}t[MAX*50];
int tot=1;
void Insert(int p)
{
	int nw=1;fr[0]=p;
	for(int i=p;i<=n&&i<=p+40;++i)
	{
		int c=ch[i]-48;
		if(!t[nw].ch[c])t[nw].ch[c]=++tot;
		nw=t[nw].ch[c];
		fr[i-p+1]=max(fr[i-p+1],t[nw].lst);
		t[nw].lst=p;
	}
}
int main()
{
	n=read();Q=read();scanf("%s",ch+1);
	for(int i=1;i<=Q;++i)L[i]=read(),R[i]=read(),A[R[i]].push_back(i);
	for(int i=1;i<=n;++i)
	{
		Insert(i);
		for(int q:A[i])
		{
			int sum=0;
			for(int j=1;j<=40;++j)
				if(fr[j]>=L[q])sum+=fr[j]-L[q]+1;
			ans[q]=sum;
		}
	}
	for(int i=1;i<=Q;++i)printf("%d\n",ans[i]);
	return 0;
}