Jzoj5462【NOIP2017提高A组冲刺11.8】好文章

nodgd写了一篇文章,自认为这是一篇好文章。nodgd的文章由n个小写英文字母组成。文章的一个子串指的是文章中的一段连续的字母,子串的长度就是这一段的字母个数。nodgd在文章中用了排比、对偶、前后照应之类的手法,所以就有很多个子串是相同或者相近的。为了向大家证明这是一篇好文章,nodgd决定给自己的文章进行评分。nodgd首先确定了一个整数m,然后统计出文章中有多少个不相同的长度为m的子串,这个数量就是文章的评分。
然而,nodgd懒得老老实实计算这个评分了,就把任务丢给了你。

这个题卡哈希,要么用双哈希,还是sam比较稳'233333',哈希就是拿来对拍的

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 400010
using namespace std;
char str[N];
int s[N][26],mx[N],sz[N],f[N];
int cnt=1,lst=1,r[N],v[N],n,m,ans=0;
int extend(char c){
	int p=lst,np=lst=++cnt,q,nq;
	c-='a'; mx[np]=mx[p]+1;
	for(;p&&!s[p][c];p=f[p]) s[p][c]=np;
	if(!p) return f[np]=1;
	q=s[p][c];
	if(mx[q]==mx[p]+1) f[np]=q;
	else {
		nq=++cnt;
		mx[nq]=mx[p]+1;
		f[nq]=f[q]; f[q]=f[np]=nq;
		memcpy(s[nq],s[q],26<<2);
		for(;p&&s[p][c]==q;p=f[p]) s[p][c]=nq;
	}
}
int main(){
	freopen("article.in","r",stdin);
	freopen("article.out","w",stdout);
	scanf("%d%d%s",&n,&m,str+1);
	for(int i=1;i<=n;++i) extend(str[i]);
	for(int i=1;i<=cnt;++i) ++v[mx[i]];
	for(int i=1;i<=n;++i) v[i]+=v[i-1];
	for(int i=cnt;i;--i) r[v[mx[i]]--]=i;
	for(int p,i=cnt;i;--i){
		p=r[i];
		if(mx[p]>=m && mx[f[p]]<m) ans++;
	}
	printf("%d\n",ans);
}

posted @ 2017-11-08 15:40  扩展的灰(Extended_Ash)  阅读(121)  评论(0编辑  收藏  举报