# 【洛谷2178】[NOI2015] 品酒大会（后缀数组+单调栈）

### 后缀数组

$i,j$$k$相似的，当且仅当$Height$数组上区间$[i,j-1]$的最小值大于等于$k$

### 代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 300000
#define LL long long
#define Gmax(x,y) (x<(y)&&(x=(y)))
#define Gmin(x,y) (x>(y)&&(x=(y)))
using namespace std;
int n,a[N+5];LL t1[N+5],t2[N+5];char s[N+5];
class FastIO
{
private:
#define FS 100000
#define pc(c) (C==E&&(clear(),0),*C++=c)
#define D isdigit(c=tc())
int f,T;char c,*A,*B,*C,*E,FI[FS],FO[FS],S[FS];
public:
I FastIO() {A=B=FI,C=FO,E=FO+FS;}
Tp I void read(Ty& x) {x=0,f=1;W(!D) f=c^'-'?1:-1;W(x=(x<<3)+(x<<1)+(c&15),D);x*=f;}
I void reads(CI n,char *s) {W(isspace(c=tc()));for(RI i=1;i<=n;++i) s[i]=c,c=tc();}
Tp I void write(Ty x) {x<0&&(pc('-'),x=-x);W(S[++T]=x%10+48,x/=10);W(T) pc(S[T--]);}
Tp I void write(Con Ty& x,Con Ty& y) {write(x),pc(' '),write(y),pc('\n');}
I void clear() {fwrite(FO,1,C-FO,stdout),C=FO;}
}F;
class SuffixArray//后缀数组
{
private:
int t[N+5],p[N+5],rk[N+5],SA[N+5],H[N+5];
I void Sort(CI S)
{
RI i;for(i=0;i<=S;++i) t[i]=0;for(i=1;i<=n;++i) ++t[rk[i]];
for(i=1;i<=S;++i) t[i]+=t[i-1];for(i=n;i;--i) SA[t[rk[p[i]]]--]=p[i];
}
I void GetSA()
{
RI i;for(i=1;i<=n;++i) rk[p[i]=i]=s[i];
RI k,t=0,S='z';for(Sort(S),k=1;t^n;k<<=1,S=t)
{
for(i=1;i<=k;++i) p[i]=n-k+i;for(t=k,i=1;i<=n;++i) SA[i]>k&&(p[++t]=SA[i]-k);
for(Sort(S),i=1;i<=n;++i) p[i]=rk[i];for(rk[SA[1]]=t=1,i=2;i<=n;++i)
(p[SA[i]]^p[SA[i-1]]||p[SA[i]+k]^p[SA[i-1]+k])&&++t,rk[SA[i]]=t;
}
}
I void GetH()
{
RI i,j,k=0;for(i=1;i<=n;++i) p[SA[i]]=i;for(i=1;i<=n;++i)
{
if(k&&--k,p[i]==1) continue;j=SA[p[i]-1];
W(i+k<=n&&j+k<=n&&s[i+k]==s[j+k]) ++k;H[p[i]]=k;
}
}
public:
I void Init() {GetSA(),GetH();}I int operator [] (CI x) Con {return H[x];}//返回Height数组
I int operator () (CI x) Con {return a[SA[x]];}//返回排名为x的位置的值
}S;
struct P
{
int k,p,l,Mx,Mn;I P(CI t=0,CI i=0,CI s=0,CI f=-1e9,CI g=1e9):k(t),p(i),l(s),Mx(f),Mn(g){}
I void operator += (Con P& x) {l+=x.l,Gmax(Mx,x.Mx),Gmin(Mn,x.Mn);}//合并区间时，更新长度和极值
}St[N+5];
int main()
{
for(S.Init(),i=0;i<=n;++i) t2[i]=-1e18;RI T=0;P k;for(i=2;i<=n;++i)
{
k=P(S[i],i,1,S(i-1),S(i-1));W(T&&St[T].k>=k.k)//k初始化为当前位置信息
Gmax(t2[St[T].k],max(1LL*St[T].Mx*k.Mx,1LL*St[T].Mn*k.Mn)),//对于要删的区间，更新第二问
t1[St[T].k]+=1LL*(i-St[T].p)*St[T].l,k+=St[T--];St[++T]=k;//更新第一问，维护新区间信息
}
for(i=T;i;--i) t1[St[i].k]+=1LL*(n-St[i].p+1)*St[i].l;//对于栈中剩余元素求解第一问
for(St[T+1]=P(0,0,0,S(n),S(n)),i=T;i;--i) Gmax(t2[St[i].k],//注意往栈中加入第n个元素
max(1LL*St[i].Mx*St[i+1].Mx,1LL*St[i].Mn*St[i+1].Mn)),St[i]+=St[i+1];//对于栈中剩余元素求解第二问
for(i=n-2;~i;--i) t1[i]+=t1[i+1],Gmax(t2[i],t2[i+1]);//统计答案
for(i=0;i^n;++i) t1[i]?F.write(t1[i],t2[i]):F.write(0,0);return F.clear(),0;
}

posted @ 2020-06-15 19:13  TheLostWeak  阅读(30)  评论(0编辑  收藏