BZOJ4199: [Noi2015]品酒大会

fail树的应用。

在fail树上找到所有长度子串的信息的话，用差分最后累加是个不错的选择。

这道题主要注意负数的处理。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<set>
#include<map>
#include<queue>
using namespace std;
#define mem1(i,j) memset(i,j,sizeof(i))
#define mem2(i,j) memcpy(i,j,sizeof(i))
#define LL long long
#define up(i,j,n) for(int i=(j);i<=(n);i++)
#define FILE "dealing"
#define poi vec
#define eps 1e-10
#define db double
const LL maxn=620000,inf=2000000000000000000LL,mod=1000000007,pp=1e8;
int read(){
    int x=0,f=1,ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0',ch=getchar();}
    return f*x;
}
inline bool cmax(LL& a,LL b){return a<b?a=b,true:false;}
inline bool cmin(LL& a,LL b){return a>b?a=b,true:false;}
int n;
char s[maxn];LL a[maxn];
int cnt=1,now=1,len[maxn],val[maxn],pre[maxn],c[maxn][26],id[maxn];
LL ans[maxn],Max[maxn],sum[maxn],Maxval[maxn],Minval[maxn],Maxans[maxn];
int extend(int x){
	LL np,nq,q,p;
	p=now;now=np=++cnt;len[np]=len[p]+1;sum[np]=1;
	while(p&&!c[p][x])c[p][x]=np,p=pre[p];
	if(!p)pre[np]=1;
	else {
		q=c[p][x];
		if(len[q]==len[p]+1)pre[np]=q;
		else {
			len[nq=++cnt]=len[p]+1;
			mem2(c[nq],c[q]);
			pre[nq]=pre[q];
			pre[q]=pre[np]=nq;
			while(p&&c[p][x]==q)c[p][x]=nq,p=pre[p];
		}
	}
	return np;
}

void build(char* s){
	cnt=now=1;
	up(i,1,n)
		id[i]=extend(s[i]-'a');
}
int cou[maxn],sa[maxn],t[maxn],y[maxn];
void getsort(){
	up(i,1,cnt)cou[len[i]]++;
	up(i,1,cnt)cou[i]+=cou[i-1];
	for(LL i=cnt;i>=1;i--)sa[cou[len[i]]--]=i;
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
    n=read();
	scanf("%s",s+1);
	reverse(s+1,s+n+1);
	for(int i=n;i>=1;i--)a[i]=read();
	build(s);
	getsort();
	up(i,0,cnt)Minval[i]=inf,Maxval[i]=-inf,Max[i]=-inf,Maxans[i]=-inf;
	up(i,1,n)Maxval[id[i]]=Minval[id[i]]=a[i];
	up(i,1,cnt)t[i]=pre[sa[i]];
	for(int i=cnt;i>=1;i--){
		sum[t[i]]+=sum[sa[i]];
		if(Maxval[t[i]]!=-inf)cmax(Max[t[i]],Maxval[t[i]]*Maxval[sa[i]]);
		if(Maxval[t[i]]!=-inf)cmax(Max[t[i]],Maxval[t[i]]*Minval[sa[i]]);
		if(Minval[t[i]]!=inf)cmax(Max[t[i]],Minval[t[i]]*Maxval[sa[i]]);
		if(Minval[t[i]]!=inf)cmax(Max[t[i]],Minval[t[i]]*Minval[sa[i]]);
		cmax(Maxval[t[i]],Maxval[sa[i]]);
		cmin(Minval[t[i]],Minval[sa[i]]);
		cmax(Maxans[len[t[i]]],Max[t[i]]);
	}
	for(int i=cnt;i>=1;i--){
		ans[len[sa[i]]]+=sum[sa[i]]*(sum[sa[i]]-1)/2;
		if(sa[i]!=1)ans[len[t[i]]]-=sum[sa[i]]*(sum[sa[i]]-1)/2;
	}
	for(int i=n;i>=0;i--)ans[i]+=ans[i+1],cmax(Maxans[i],Maxans[i+1]);
	up(i,0,n-1)printf("%lld %lld\n",ans[i],(Maxans[i]==-inf)?0:Maxans[i]);
	
    return 0;
}