SPOJ - DISUBSTR

最新推荐文章于 2020-03-31 22:56:17 发布

二分抄代码

最新推荐文章于 2020-03-31 22:56:17 发布

阅读量279

点赞数

分类专栏：后缀自动机文章标签：后缀自动机

本文链接：https://blog.csdn.net/liufengwei1/article/details/78013183

版权

后缀自动机专栏收录该内容

2 篇文章 0 订阅

订阅专栏

这题体现了后缀自动机的不足之处，如果此题长度是什么10^5，然后又不告诉你有多少种字符的话，不用后缀数组而去用后缀自动机就会炸空间。不过此题还是可做的，对于每一个结点所在的位置为右节点的后缀可以用st[i].len-st[st[i].pre].len;短于等于st[i].pre前面的之后到了那个结点又会继续统计。

#include<cstdio>
#include<cstring>
#define maxl 1010

struct sam
{
	int pre,len;
	int son[256];
}st[maxl<<1];
int slen,sz,last,ans;
char s[maxl];

void prework()
{
	scanf("%s",s+1);
	slen=strlen(s+1);
	sz=0;last=0;
	for(int i=0;i<=slen<<1;i++)
	{
		st[i].len=0;st[i].pre=0;
		for(int j=0;j<256;j++)
			st[i].son[j]=0;
	}
	st[0].pre=-1;st[0].len=0;++sz;
}

void extend(char cc)
{
	int cur=sz++,c=cc;
	st[cur].len=st[last].len+1;
	int p;
	for(p=last;p!=-1 && !st[p].son[c];p=st[p].pre)
		st[p].son[c]=cur;
	if(p==-1)
		st[cur].pre=0;
	else
	{
		int q=st[p].son[c];
		if(st[p].len+1==st[q].len)
			st[cur].pre=q;
		else
		{
			int clone=sz++;
			st[clone].len=st[p].len+1;
			for(int j=0;j<256;j++)
				st[clone].son[j]=st[q].son[j];
			st[clone].pre=st[q].pre;
			for(;p!=-1 && st[p].son[c]==q;p=st[p].pre)
				st[p].son[c]=clone;
			st[q].pre=clone;st[cur].pre=clone;
		}
	}
	last=cur;
}

void mainwork()
{
	for(int i=1;i<=slen;i++)
		extend(s[i]);
	ans=0;
	for(int i=1;i<=sz;i++)
		ans+=st[i].len-st[st[i].pre].len;
}

void print()
{
	printf("%d\n",ans);
}

int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}