P4248 [AHOI2013]差异 (后缀数组+单调栈)
题目链接:传送门
正文:
首先对于这个公式的前两项我们可以快速求出,为 ( 1 + n ) ∗ n ∗ ( n − 1 ) / 2 (1+n)*n*(n-1)/2 (1+n)∗n∗(n−1)/2。所以我们只需考虑最后一项的和,又因为每个 T i T_i Ti对应排序后的后缀的 r a n k [ i ] rank[i] rank[i]位置,所以这步可以转化为在排序后的后缀数组上统计答案。
首先最后一项 l c p ( T i , T j ) lcp(T_i,T_j) lcp(Ti,Tj) 的值,设 x = r a n k [ i ] , y = r a n k [ j ] x=rank[i],y=rank[j] x=rank[i],y=rank[j],不访假设 x < y x<y x<y,那么 l c p ( T i , T j ) lcp(T_i,T_j) lcp(Ti,Tj)等于 m i n { h e i g h t [ x + 1 ] , h e i g h t [ x + 2 ] . . . . , h e i g h t [ y ] } min\{height[x+1],height[x+2]....,height[y]\} min{height[x+1],height[x+2]....,height[y]},即区间 [ x + 1 , y ] [x+1,y] [x+1,y]的 h e i g h t [ ] height[] height[]数组中的最小的值 h e i g h t [ k ] height[k] height[k] 的值是贡献。如果最小高度有多个,那么对规定最左边的最小高度起贡献。
因为最后一项的是所有元组,所以我们可以用单调栈维护每个 h e i g h t [ ] height[] height[]贡献的区间即可,
代码:
#include<bits/stdc++.h>
#define mset(a,b) memset(a,b,sizeof(a))
typedef long long ll;
const int N=5e5+10;//下标从1开始
int rak[N],sa[N],tp[N],c[N],height[N];
//最后的rak[i]=p 与sa[p]=i ,11对应
void SA(char *s,int n)//这个基数排序的版本中桶的大小为第一关键词rak的最大值
{
int m=128;//桶的大小,会慢慢变大,最大为n,但初始时是字符的最大值
//首先基数排序初始化rak数组和sa数组
//第一轮基数排序,如果s的最大值很大,可改用快速排序
for(int i=0; i<=m; ++i) c[i]=0;
for(int i=1; i<=n; ++i) c[rak[i]=s[i]]++;
for(int i=1; i<=m; ++i) c[i]+=c[i-1];
for(int i=n; i>=1; --i) sa[c[s[i]]--]=i; //所有后缀长度为1的字符串已求出
for(int k=1,p ; k < n; k<<=1 ) //p是一个计数器,现在还没用。
{
/*tp[i]:第二关键词排名为i的位置为tp[i]*/
p=0;
for(int i=n-k+1; i<=n; ++i) tp[++p]=i;
for(int i=1; i<=n; ++i) if(sa[i]>k) tp[++p]=sa[i]-k;
/*基数排序求出2k长度的sa数组*/
for(int i=0; i<=m; ++i) c[i]=0;
for(int i=1; i<=n; ++i) c[rak[i]]++;
for(int i=1; i<=m; ++i) c[i]+=c[i-1];
for(int i=n; i>=1; --i) sa[ c[rak[tp[i]]]-- ]=tp[i];
/*求该轮的rank数组*/
//tp才是上一轮的rak,现在要利用上轮的rak和这轮的sa求这轮的rank
std::swap(rak,tp);
rak[sa[1]]=p=1;
for(int i=2,a,b; i<=n; ++i)
{
a=sa[i],b=sa[i-1];
if(tp[a]==tp[b]&&((a+k<=n&& b+k<=n&&tp[a+k]==tp[b+k])||(a+k > n&& b+k > n))) rak[a]=p;
else rak[a]=++p;
}
if(p>=n) break;//可以提前退出
m=p;
}//计算高度
int k=0;
for(int i=1; i<=n; ++i)
{
if(k) k--;//i ,
int j=sa[rak[i]-1];
if(j==0) continue;
while(s[j+k]==s[i+k]) ++k;
height[rak[i]]=k;
}
}
char s[N];
int S[N],top;
int left[N],right[N];
int main()
{
scanf("%s",s+1);
int ls=strlen(s+1);
SA(s,ls);
top=0;
for(int i=2;i<=ls;++i)
{
while(top > 0&& height[S[top]]> height[i]) top--;
S[++top]=i;
if(top==1) left[i]=2;
else left[i]=S[top-1]+1;
}
top=0;
for(int i=ls;i>=2;--i)
{
while(top>0 && height[S[top]] >=height[i]) top--;
S[++top]=i;
if(top==1) right[i]=ls;
else right[i]=S[top-1]-1;
}
long long ans=(1ll+ls)*(ls-1)*(ls)/2ll;
for(int i=2;i<=ls;++i)
{
ans-=2ll*height[i]*(right[i]-i+1)*(i-left[i]+1);
}
printf("%lld\n",ans);
}