Oxer的专栏

已退役

【bzoj4199】[Noi2015]品酒大会 后缀自动机

听说对反串建SAM,fa树就是后缀树?

听说两个后缀的LCP就是LCA的mx?

然后就是个树上dp?

靠,老子边界打错了


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 600010

using namespace std;

int head[maxn],next[2*maxn],to[2*maxn];
int ch[maxn][26],fa[maxn],mx[maxn],a[maxn],w[maxn],size[maxn];
char s[maxn];
long long cnt[maxn],ans[maxn];
long long MX[maxn],MN[maxn];
int n,m,num,tot=1,last=1;

void insert(int x,int W)
{
	int p=last,np=last=++tot;
	mx[np]=mx[p]+1;w[np]=W;
	while (p && !ch[p][x]) ch[p][x]=np,p=fa[p];
	if (!p) fa[np]=1;
	else
	{
		int q=ch[p][x];
		if (mx[q]==mx[p]+1) fa[np]=q;
		else
		{
			int nq=++tot;w[nq]=1e9+1;mx[nq]=mx[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[nq]=fa[q];
			fa[np]=fa[q]=nq;
			while (p && ch[p][x]==q) ch[p][x]=nq,p=fa[p];
		}
	}
}

void addedge(int x,int y) {num++;to[num]=y;next[num]=head[x];head[x]=num;}

void dfs(int x)
{
	MX[x]=-1e9-1;MN[x]=1e9+1;size[x]=0;
	if (w[x]!=1e9+1) size[x]=1,MX[x]=MN[x]=w[x];
	for (int p=head[x];p;p=next[p])
	{
		dfs(to[p]);
		if (MX[x]!=-1e9-1 && MN[x]!=1e9+1 && MX[to[p]]!=-1e9-1 && MN[to[p]]!=1e9+1) 
		  ans[mx[x]]=max(ans[mx[x]],max(MX[x]*MX[to[p]],MN[x]*MN[to[p]]));
		MX[x]=max(MX[x],MX[to[p]]);MN[x]=min(MN[x],MN[to[p]]);
		cnt[mx[x]]+=1ll*size[x]*size[to[p]];size[x]+=size[to[p]];
	}
}

int main()
{
	scanf("%d",&n);
	scanf("%s",s+1);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	for (int i=n;i>=1;i--) insert(s[i]-'a',a[i]);
	for (int i=1;i<=tot;i++) addedge(fa[i],i);
	for (int i=0;i<=n;i++) ans[i]=-1e18;
	w[1]=1e9+1;
	dfs(1);
	for (int i=n-1;i>=0;i--) cnt[i]+=cnt[i+1],ans[i]=max(ans[i],ans[i+1]);
	for (int i=0;i<=n-1;i++) if (cnt[i]) printf("%lld %lld\n",cnt[i],ans[i]); else printf("0 0\n");
	return 0;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u012288458/article/details/51793280
个人分类: 字符串
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

【bzoj4199】[Noi2015]品酒大会 后缀自动机

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭