Jzoj4734 恶魔

其实就是在求有几对完全相等子串满足长度小于等于k而已

很明显打一个sam就好了，条件就是mx[f[p]]<k就可以计算贡献

#include<stdio.h>#include<string.h>#include<algorithm>#define N 200010 using namespace std;char str[N]; long long A=0;int s[N][26],mx[N],f[N],sz[N];int n,k,cnt=1,lst=1,v[N],r[N];long long C(int x){ return x*(x-1ll)>>1; } int extend(char c){int p=lst,np=lst=++cnt,q,nq;c-='a'; mx[np]=mx[p]+1; sz[np]=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(){scanf("%d%d%s",&n,&k,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]; sz[f[p]]+=sz[p];if(mx[f[p]]<k) A+=C(sz[p])*(min(k,mx[p])-mx[f[p]]);}printf("%lld\n",A);}