题目
给定一个长度为(n)的串，定义两个串匹配当且仅当两个串长度相同并且不同字符至多一个
对于每一个长度为(m)的子串输出和它匹配的子串个数
$1 le n le 10^5 , m le n $ 字符集=4
题解
- 我不知道字符集为什么等于4..
- 匹配的条件相当于$lcp+lcs ge m-1 $
- 正反建(SA)，倒着枚举(lcp)的值，相当于合并(suffix)
- 要求合法的(lcs)大于等于(m-1-lcp)，每次合并在(preffix)上二分贡献答案
- 用线段树维护答案
- 实现可以启发式+线段树合并
- 时间复杂度 (O(n log ^2n))
#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define mk make_pair
#define fi first
#define se second
#define il inline
using namespace std;
const int N=100010,M=200;
int n,m,bin[17],sum[N*M],add[N*M],ls[N*M],rs[N*M],fa[N],sz[N],cnt,st[N],rt[N],lg[N];
vector<int>vec[N];
char s[N];
ll ans[N];
struct SA{
int sa[N],rk[N],ht[N],f[N][17];
void init(){
//printf("%s
",s);
build_sa();
//for(int i=0;i<n;++i)printf("% 2d",sa[i]);puts("");
build_ht();
//for(int i=0;i<n;++i)printf("% 2d",ht[i]);puts("");
build_rmq();
}
void build_sa(){
int m=128;
static int x[N],y[N],c[N];
for(int i=0;i<m;++i)c[i]=0;
for(int i=0;i<n;++i)c[x[i]=s[i]]++;
for(int i=1;i<m;++i)c[i]+=c[i-1];
for(int i=n-1;~i;--i)sa[--c[x[i]]]=i;
for(int k=1,p=0;k<n&&p<n;k<<=1,m=p){
p=0;for(int j=n-k;j<n;++j)y[p++]=j;
for(int i=0;i<n;++i)if(sa[i]>=k)y[p++]=sa[i]-k;
for(int i=0;i<m;++i)c[i]=0;
for(int i=0;i<n;++i)c[x[i]]++;
for(int i=1;i<m;++i)c[i]+=c[i-1];
for(int i=n-1;~i;--i)sa[--c[x[y[i]]]]=y[i];
p=1;swap(x,y);x[sa[0]]=0;
for(int i=1;i<n;++i){
x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
}
}
}
void build_ht(){
for(int i=0;i<n;++i)rk[sa[i]]=i;
for(int i=0,j=0,k=0;i<n-1;ht[rk[i++]]=k){
j=sa[rk[i]-1];if(k)k--;
while(s[j+k]==s[i+k])++k;
}
}
il int min(int x,int y){if(x<y)return x;return y;}
void build_rmq(){
for(int i=1;i<n;++i)f[i][0]=ht[i];
for(int i=1;i<17;++i)
for(int j=1;j+bin[i]-1<n;++j){
f[j][i]=min(f[j][i-1],f[j+bin[i-1]][i-1]);
}
}
il int ask(int l,int r){
if(l==r)return n+1;
int t=lg[r-l];
return min(f[l+1][t],f[r-bin[t]+1][t]);
}
void find(int x,int y,int&L,int&R){
x=rk[x];
int l=0,r=x;
while(l<r){
int mid=(l+r)>>1;
if(ask(mid,x)>=y)r=mid;
else l=mid+1;
}L=l;
l=x,r=n;
while(l<r){
int mid=(l+r+1)>>1;
if(ask(x,mid)>=y)l=mid;
else r=mid-1;
}R=r;
}
}pre,suf;
bool cmp(int a,int b){return suf.ht[a]<suf.ht[b];}
void pushdown(int k){
int l=ls[k],r=rs[k];
if(l)add[l]+=add[k];
if(r)add[r]+=add[k];
add[k]=0;
}
void ins(int&k,int l,int r,int x){
sum[k=++cnt]++;
if(l==r)return;
int mid=(l+r)>>1;
if(x<=mid)ins(ls[k],l,mid,x);
else ins(rs[k],mid+1,r,x);
}
int modify(int k,int l,int r,int x,int y){
if(!k)return 0;
if(l==x&&r==y){add[k]++;return sum[k];}
int mid=(l+r)>>1;
if(y<=mid)return modify(ls[k],l,mid,x,y);
else if(x>mid)return modify(rs[k],mid+1,r,x,y);
else return modify(ls[k],l,mid,x,mid)+modify(rs[k],mid+1,r,mid+1,y);
}
void getans(int k,int l,int r){
if(!k)return;
if(l==r){
ans[n-m-pre.sa[l]]+=add[k];
return;
}
if(add[k])pushdown(k);
int mid=(l+r)>>1;
getans(ls[k],l,mid);
getans(rs[k],mid+1,r);
}
int merge(int x,int y){
if(!x||!y)return x+y;
if(add[x])pushdown(x);
if(add[y])pushdown(y);
sum[x]+=sum[y];
ls[x]=merge(ls[x],ls[y]);
rs[x]=merge(rs[x],rs[y]);
return x;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void comb(int x,int y,int lim){
x=find(x),y=find(y);
if(sz[x]>sz[y])swap(x,y);
for(int i=0,l,r;i<(int)vec[x].size();++i){
int t=vec[x][i];vec[y].pb(t);
if(t>n-m)continue;
pre.find(n-m-t,lim,l,r);
ans[t]+=modify(rt[y],0,n,l,r);
}
rt[y]=merge(rt[x],rt[y]);
fa[x]=y;sz[y]+=sz[x];
}
int main(){
freopen("lovely.in","r",stdin);
freopen("lovely.out","w",stdout);
scanf("%d%d%s",&n,&m,s);
for(int i=bin[0]=1;i<17;++i)bin[i]=bin[i-1]<<1;
for(int i=2;i<=n;++i)lg[i]=lg[i>>1]+1;
s[n++]='$';suf.init();
reverse(s,s+n-1);pre.init();
n--;
for(int i=0;i<n;++i){
fa[i]=i;sz[i]=1;vec[i].pb(i);
if(i>n-m)continue;
ins(rt[i],0,n,pre.rk[n-m-i]);
}
for(int i=2;i<=n;++i)st[i]=i;
sort(st+2,st+n+1,cmp);
for(int i=m-1,j=n;~i;--i){
while(j>1&&suf.ht[st[j]]>=i){
int x=st[j--],y=x-1;
comb(suf.sa[x],suf.sa[y],m-1-i);
}
}
getans(rt[find(0)],0,n);
for(int i=0;i<=n-m;++i)printf("%lld ",ans[i]);
return 0;
}