[bzoj3238][后缀数组][ST表]差异

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Rose_max/article/details/82784992

Description

在这里插入图片描述

Input

一行,一个字符串S

Output

一行,一个整数,表示所求值

Sample Input

cacao

Sample Output

54

HINT

2<=N<=500000,S由小写英文字母组成

题解

预处理height
预处理一个ST表
考虑一个位置的height能做的贡献
显然是这个位置往前往后连续的一段
二分段
但是可能会出现一个问题
height成了
…3…3…
中间都是比3大的数
那么这样如果两边都允许相等 会出现重复的贡献被计算
强行让左边不等于 右边允许等于就可以了.

/*
:    .                                                                                                                  
:                                                                                                                       
i                                                                                                                       
i    .                                                                                                                  
i                                                                                                                       
:    .                                                                                                                  
i    .                                                                                                                  
i    .                                                                                                                  
i                                                                                                                       
:        .                                                                                                              
i  .  i  i :     . .    .   .     .       . .           ..    .                                   .          .          
.     .  . .       .                      .   .      .                                                          .       
:   qQ  .   v2d   .  i: ..:ij.iBE .i            EBj        .irrivri.::iiiir7r.                       .    SPd.  :       
i  UQB    .BBBBE     BBBQBBBBLiBB.1BBi    :X:   BBu    .   .KBBBBBBBBBBBBBBBQv                          :BBBBB          
. iBY PD. BBY vBBv.  ..:gB1   rBM  7B1    RQP   MBgvPPQBB          uRq.        ..                 .    jBBQ.BBBr        
:.BBrQBBBBBL   iQBB7:777gBSYvrKBBv:v71r..YPBBBBBBBBBBEBBB    7RBZQdBBBdEgBQE                         1QBB1   QBBB:      
.:BBBBJ.qBBDBBBBBBBvrQBQBBBBBBBBBBBBBgg5UBBBBriiBQ1   rBB  .:.:i. :KQBi ..::    :.. . . .   .....  UBBB5  UJ: rBBBBXi   
.  IQY     i:.:..       YBs   :BB   IE.   XBZ   QBs   7BB  rBBBBQBBBBBBBQBBBBq BQBBBBBBBBBBBBBBBBviBBE.  .BBZ   .BBgi.  
. BBBQSJ.Kb5BQgRBPv :5gQBBBQBB:YBvrBBd    QBE   BBdiBBBB5      .BB     uBs     iririiiriir7iii77i. ii    .BBD  .        
. uQ5Sv .BB::i::QBU  BBQMBK.   7BBBP.     DBq   QB1 DQ7.L. sBBdQBBMBBBQBBBBBQDi                           BB5           
.    ...:BB     KBJ     uB1   UBBBL  QBL  RQb   i:.     BBPvi.UBBuY7v7iQBqrrur.                           BBE   .       
:LBBBBBBJQBBBBQBBBJ  :rvBBu:BBB1jBBPbBB   SBBZIvSUguJXRBBB:  7QBB      BBu     ..                         BBQ           
r.XX... .QBIYXDJBB1  qBBBBi ir   rMQBQ.   .sPBBBBBBBBRBBD7  XBBi       dB1. .  ..                        .BBg           
i                .                   .                           . . .         ..                         ... .         
:                                                                                                                       
i    .                                                                   .                                              
i                                                                                                                       
i    .                                                                                                                  
i                                                                         :r:.                                          
i    .                                                                .  vKKSj. .                                       
r                                                                 ..    :B: js  .                                       
i    .                                                             .    rK .S:                 :::                      
r                                                                       :v LS             .  ibi7gv   .                 
i                                                                       :Ligi               .QJ :g7  .                  
.      ...5d:  .                                                        i2Sd               rb. vj:      .               
.   .7:r27DBi                                                      .    .Bb:     .        7K..Ps                        
uij1Pdr.  ri                                                       .     Sq              :Kj7Ii .  .     :v:            
UvY:     7P. .                                                     .     .7             .vIqX. ..  .    .dBRS7.         
.    .   qP     . .       .  ..                                       .     .  ..        :r.            .qi :2ZPi       
.    .   gs     .    .:K2  .          .irvIs75K5XEKMKgZsD522YLri.              iBB1i                     Pr   .2r       
i       iS:        .isjjQ.    .:rs5KPPMBBBZdBBgi:7Rd7XBBBRSMBBDZQZSr.        rDYiiYdQSr.                 dY             
:       Ys      .iXbi.  7: 7v5gQZqPEKXUIjUIdbr   ur    jRQr1E:  .rJgBBDL:   iEv     .sdD5Yi.             MX  .   .      
i       Si   .vU57i   .ibQBBgMPI175XUvsq5LBj   :Li.     jQLJ:  .    .rdBBQsYL           :r1ISv:         .gu    .        
:   .. iU  r5Qu.   :sBBD7qDqbbijPj5ISXv22UK   .BY .     .gSP: .q:      rBBBBS    .   .      .7KEYi.      qv    .        
.      qBsY1i   :IqBZqI2rUXLXQPrYjIuED2jZPv.  KI .  .    .BR. rd        .bBUBQu.          .    .isZK1:   5s             
:      vPv.  .idDBEP1XYKXQR1r5d2SdPqEDv7IEY  Yv    .      KB.:U.  :Ui    .BJUBBB1        .         rXqIrrB2    .        
v           rQDM1dj7SM1qIuPgXJKS2sKU5urLDMr :Bi   .  .dBMi7BPYJ 1PKL:     BjKIUPBB1       .       .   :irMX             
i  .     .vKDDr1YREIPgv1qS1Qbju1IS2US5vUZBu sr .    :BBBBdXMDMgjv.   :    guJsP7rQgU:                    JdrsSv:        
7       rQREJ5u2vuKv5dXKKPXqqXK1PSSqP2USqqBDj    .  7QBQRPQriQg:..v7i:.   MqEXBs72DQB:              .iYYr..iPRJ         
7  .  .MRB221QqKrYSSXdXXuBZuSguL1sPMDQQDPPgBi    .  :RBBMB5  PELBBBj     vP.  irRBgRPEv           .Ju7ivv7vLi.   .      
:    7DQUdSPqgIqUsKSYv5Si2PrruJSBZR5r::   :Z:        .irvX::qPPZMBBQgs  .Xs      LPuXBBi         :Dd2rLsr.      .       
i   YBZ7Lg2bUPLb172Xr7PQvJuuuMBbX7.       .uRu.         :QEBQRRBZgggQBSrY.         :SQQX         :27:.       .          
i .EP2PX1qJYIPuu5I2g2uB5YXbDZPJ   .rr.  .   7gZU7r7vYvvvJqBQRBMdgDMgBQQr            r5BBr                               
 .gB2LgEsPPuKE2IrM5Iu1U2JXgS.     :7vvvsYLL7. .rs77JJr.  JBQDBbdEZBRgBP      . .YIUvirvBQ:   .                          
XQQSqgDJ7qg5bPXqLd1bQ2JQB2.              ..:7Lv7:        iBQSREEBQQRBP:     iSPL7.     rBu                              
QgUjEE2qJbKsjX2duqYKr5BS.      ..             .rYr.       dBQBZBQBKgM. .  rZ2:          jEi                             
vJdPIJI5UPSYKb5uLbLSDEu       .                            :BBgBPEBr  .  ....         . iBK.                            
YvLb7K5L7ubsjgJuuSsPQ7  .....:iiirr77v7vrv1PXU77rr::.            :E.  .           .. ..  sgi                            
LuKQrsSd5XQPqSij7bBu    :ivu5J7i::::::::::i::.::iri..            :Ei.     ..i715qLPdUqS. iBi     .                      
7UPBv2IPqP1IdXLZZPr   .                                .         .sY       .ir:.     ..  .QE  ..                        
UsJRubSv7ggUJr1Qg:                             ..                 r5                      vd.                           
d5YPrSqX7QIvPRBX.  .  .            ..:i777r7vJvr:      .         .rZ  .                 .iE5rr                          
2LKXj2E17DjSMSQi          ..ii7uqS522J7:.....              ..     7Z.    .  .75Y::.  7KdBZEXXBQ:                        
r7PQqdEjJEPJuZJ       iIPK1j7:.:.         .:i:......              :P        .iLuvPSsXQbr     .JBP                       
sLPP1uqs2XR7JQ:       .:i.        .:71ZBBBQEDDMbdMBBBdui:         rZ.               .          bB.                      
UUqK7vSIv5PuDg                .YdQBBBBQBMESJ777s1bPdEgRQBBRUi.    .5       .                  .gq                       
vLZPvvbKLdSIBv          .   1MBRQBBBMEbZPvgBRJrLY7Uj2J7rvPDQBBgUY722            ...       ..  iBJ                       
irIKvUPI7M1SRi     .  .   7BB1JUvSd1Y512IYSBMKLPv1XirsivvUKgDgBBBBBBKr:.                  ..  uE.                       
D2KIJ5D17EPdd.  .        qBJv.ivivYi7JvrLrLEgbbP1PE:71sS5sDq7vDQD2qbgBBBMJi                  rR7                        
vJPbvJq2YgEQ2           .BP.L7ijIEuiivr7vii1PKgg5dRjXgUSujJ7JgJ7Piru1UKMBBQM2i             .2I:.                        
vrXP1Xg2vJjbL    .  .   sBBuXI1gBgE2SI7vIrrsIiSISdBSEZ1bggQv7Es:U77YLirugggDBBBgKUJrri:.:7IEP.                          
XU1UubPuLu7LY    .     :5RBgUBBPPdggBEUdqIKXviuJrYdPPq2PKKBqY1jrj75I1LL7UXZBBdPEQBP7jIIdKSBB.  .                        
sIDbX5X5KRBSE         .ZXLXI7q57UJ5IRUUZP5KZguivSv1S55bS12BMBjJ7dXBgZIqr7dqMEPBEQQ:       KB7         ...       ..      
PU7v5MPJ7KZPD         iBbU7v77:r7irYdIud5KqZPsrs2vYUsJuPP5SLKgQdEEgdPgMX7qvI5DXbMX       :BQ                            
511sSdZXI51vg.     .  LgXgU77iir5dddBMMBBBP5dP52S2gBQ2vL22RdBQBPQP15SdMR1Jivrr:MBr       SBv   .jDBMbuv:      ..        
vU52sujjUEq1E.   . .  2MrBDU5PQBZP7.:..irDQQPdBBRqi.      .:jXQMBPrIjXXBP1iuii.BE       .BX  rSQK:...i1dX7.             
KJ1svuqsrEgPK      .  v1rDQDBj:.          .LQBS.              .JBE.r7555RbbK55PBr    .  Kg  7QM.        7BBv            
PXdEu5PsLKv5di        7E1dKi                :J.                 :gg1iq1LsQRRYuBP   .r  KQi iBg           .2Q5.  .       
IjPP1ID5UJYdBY        rBBS    .                 .                :BErj7YU5ZPdBZ       vBv :Rd              IBi          
2UKK7JPXuIgQBb    .    bBi     .         .   .           .        EBPYv7srIKBB       vBr  sBi    .         .B1          
vvbb15d5bSPLPB:   .    rB:    ..          .                       :qBKQvYLJRK       iQ2 :ZBBr   .           Rv .        
SY57vDIKZLuuKBi        .BB.    .   .                             . bBXgudD5v      .2BBgqqbgBJ   .    ...   .E5          
v75vYqPPq5KPjJE:  .     iBBr    .                                 .BMvBQEv:       BBq1EsSQS5Bi  .   .      :BL          
vUUs7YuJrXJd7:BP     .   .PQqv:                          ..  .:77juBQJsi        rBQYiJU1JX5PRb:           vBL    .      
dZsY55RS1qjdqrBBL          .JPMRP77.   .:.....:.:i7vjuSqEEbXqqquL7. .         .jBQJvdR5vuuuJ5BBq:       isBB  . .       
BQ7v1KDu5qYSqsJgQ:             .r7IEPUs11uj25X21JJri::...                    7QB1rPQY2duIQKXZJ7BBdu7:isDZ5:    ..       
LYRgsvSUPqX1UUI2Bd.                                                       ii2MEgI7dRSJ1sSBX7SPXsPZBBXr77.               
 vBB1PBZMQggZDZMXgBBDgMRbbEQQRZPZRPKSPbDRBDKujsuYv25Ys2qSEPDMMddZgZMQBBBgQBQEPsqXJSqsIP52M5UPPsvERZi                    
s5QgPQBIDMMDgDgDgBBgEBBBBBBQQBBBBQBBBBBQQBBBBBBBBQBBBBBQBBBBBBBBBQBBBBBBBMBQgESDDSEKUPgU2qSqUrPBBv.   .                 
rrPBBgMDMQRRMQgQDMQBQBMgRQggDgRBBBgZgBBBQBBBBBBBBBBBBBS:     ..vbBBBQBQMbSQQMEYSuLI5uPKv1D77PBBXi      ...              
i LdMdKKqqPgZSgd1PZPSLXPXSZZXPZIQZJ5EKdDgMMdqYr.::.:bB:          LBXIQBKPgBquP2PZuUuUSXj5ZqXBB7                         
i:BP2j:vYs7q5sZI:2uvv7Ss7UYKs51r51rJLjDg7:          .BvPPgX1vrii:2Qi  vBbIMQuZrsDP55uu2XYdBQU:   .                      
rPBLJ27PXu7SP2QIvSJsPKgYPPisXq1Lg51ZBdr             2Q: .. .qBU.:vbv.  :IQBqiSs2KYLudKv7QQb.                            
BBXr7P1PXXsPSjD5757YPPE1JJiXPZj7sSQK:                1v     iBu   g7 .   iDBPQsIPuIP7vZB1.                       .      
MuiK7uKKj11Pj:S1UbuJXv2JJSKKLU2rSgX  .....:::...::.: .Bd..i.:bj .SB7 .::  KBgMv1gu2KqDBZ:  .:ii:ri::iiirrrviii7i.        
*/
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
char ch[500005];
int Rank[500005],Rsort[500005],a[500005],sa1[500005],sa2[500005],tt[500005];
void get_sa(int len,int m)
{
	memcpy(Rank,a,sizeof(Rank));
	memset(Rsort,0,sizeof(Rsort));
	for(int i=1;i<=len;i++)Rsort[Rank[i]]++;
	for(int i=1;i<=m;i++)Rsort[i]+=Rsort[i-1];
	for(int i=len;i>=1;i--)sa1[Rsort[Rank[i]]--]=i;
	int ln=1,p=0;
	while(p<len)
	{
		int k=0;
		for(int i=len-ln+1;i<=len;i++)sa2[++k]=i;
		for(int i=1;i<=len;i++)if(sa1[i]-ln>0)sa2[++k]=sa1[i]-ln;
		memset(Rsort,0,sizeof(Rsort));
		for(int i=1;i<=len;i++)Rsort[Rank[i]]++;
		for(int i=1;i<=m;i++)Rsort[i]+=Rsort[i-1];
		for(int i=len;i>=1;i--)sa1[Rsort[Rank[sa2[i]]]--]=sa2[i];
		memcpy(tt,Rank,sizeof(tt));
		Rank[sa1[1]]=1;p=1;
		for(int i=2;i<=len;i++)
		{
			if(tt[sa1[i]]!=tt[sa1[i-1]]||tt[sa1[i]+ln]!=tt[sa1[i-1]+ln])p++;
			Rank[sa1[i]]=p;
		}
		ln*=2;m=p;
	}
}
int height[500005];
void get_he(int len)
{
	int j,k=0;
	for(int i=1;i<=len;i++)
	{
		j=sa1[Rank[i]-1];
		if(k)k--;
		while(ch[i+k]==ch[j+k])k++;
		height[Rank[i]]=k;
	}
}
int mn[500005][25],bin[25],Log[500005];
int getmn(int l,int r)
{
	int K=Log[r-l+1];
	return min(mn[l][K],mn[r-bin[K]+1][K]);
}
int main()
{
//	freopen("a.in","r",stdin);
//	freopen("a.out","w",stdout);
	Log[1]=0;for(int i=2;i<=500000;i++)Log[i]=Log[i>>1]+1;
	bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]<<1;
	scanf("%s",ch+1);int len=strlen(ch+1);
	for(int i=1;i<=len;i++)a[i]=ch[i]-'a'+1;
	get_sa(len,26);
	get_he(len);
	LL ans=((LL)(1+len))*((LL)(len))/2*((LL)len-1);
	//height[1]=999999999;
	for(int i=1;i<=len;i++)mn[i][0]=height[i];
	for(int i=1;bin[i]<=len;i++)
		for(int j=1;j+bin[i]-1<=len;j++)
			mn[j][i]=min(mn[j][i-1],mn[j+bin[i-1]][i-1]);
	for(int i=2;i<=len;i++)
	{
		int l=2,r=i,fa=i,fb=i;
		while(l<=r)
		{
			int mid=(l+r)/2;
			if(getmn(mid,i-1)>height[i])fa=mid,r=mid-1;
			else l=mid+1;
		}//左边不越界 
		fa--;
		l=i,r=len;
		while(l<=r)
		{
			int mid=(l+r)/2;
			if(getmn(i,mid)>=height[i])fb=mid,l=mid+1;
			else r=mid-1;
		}//右边越界 
		ans-=2LL*(LL)height[i]*((LL)(i-1)-fa+1)*((LL)fb-i+1);
	}
	printf("%lld\n",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Rose_max/article/details/82784992
今日推荐