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);
}