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

来源:互联网 发布:网络活动策划书籍 编辑:程序博客网 时间:2024/05/22 11:34

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

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

#include<stdio.h>#include<string.h>#include<algorithm>#define N 400010using 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);}