[BZOJ4516] [SDOI2016] 生成魔咒 - 后缀数组/后缀自动机

感觉还是SAM比较好写QAQ。

显然ans就是∑len[i]-len[fail[i]]

#include"bits/stdc++.h"
#define mmap map<int,int>::iterator
using namespace std;

const int N=200005;

struct SAM{
	map<int,int>e[N];long long res;
	int len[N],fail[N],cnt,last,n;
	SAM(){cnt=last=1;}
	int extend(int c){
		int p=last,np=++cnt;len[np]=len[p]+1;
		for(;e[p].find(c)==e[p].end()&&p;p=fail[p])
			e[p].insert(make_pair(c,np));
		if(!p)fail[np]=1;
		else {
			int q=e[p].find(c)->second,nq;
			if(len[q]==len[p]+1)fail[np]=q;
			else {
				nq=++cnt;len[nq]=len[p]+1;e[nq]=e[q];
				fail[nq]=fail[q],fail[q]=fail[np]=nq;
				for(mmap i;i=e[p].find(c),i->second==q;p=fail[p])
					i->second=nq;
			}
		}
		last=np;
		return len[np]-len[fail[np]];
	}
	void work(){
		scanf("%d",&n);int a,i;
		for(i=1;i<=n;i++){
			scanf("%d",&a);
			res+=extend(a);
			printf("%lld\n",res);
		}
	}
} sam;

int main(){
	sam.work();
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值