传送门
时隔多年(其实也就几天 )我终于把这道题A了
就是用普通的后缀数组结合单调栈,加了个 > k >k >k的限制,和[HAOI2016]找相同字符思路差不多,但有一些细节比如 h [ i ] h[i] h[i]是和上一名的 l c p lcp lcp,所以有些判断要 − 1 -1 −1之类的
一直被多组数据坑··· s [ n + 1 ] s[n+1] s[n+1]一定要置 0 0 0
代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define LL long long
#define maxn 400005
using namespace std;
int n,k,m,tax[maxn],rk[maxn],sa[maxn],h[maxn],st[maxn][20],tp[maxn];
int stk[maxn],top,f[maxn];
char s[maxn],a[maxn];
inline void rsort(){
for(int i=1;i<=m;i++) tax[i]=0;
for(int i=1;i<=n;i++) tax[rk[i]]++;
for(int i=1;i<=m;i++) tax[i]+=tax[i-1];
for(int i=n;i;i--) sa[tax[rk[tp[i]]]--]=tp[i];
}
inline void ssort(){
for(int i=1;i<=n;i++) rk[i]=s[i],tp[i]=i;
rsort();
for(int w=1,p=0;w<=n&&p<n;w<<=1,m=p){
p=0;
for(int i=n-w+1;i<=n;i++) tp[++p]=i;
for(int i=1;i<=n;i++)
if(sa[i]>w) tp[++p]=sa[i]-w;
rsort(); swap(rk,tp);
rk[sa[1]]=p=1;
for(int i=2;i<=n;i++)
if(tp[sa[i]]==tp[sa[i-1]] && tp[min(n+1,sa[i]+w)]==tp[min(n+1,sa[i-1]+w)])
rk[sa[i]]=p;
else rk[sa[i]]=++p;
}
}
inline void get_h(){
int j,k=0;
for(int i=1;i<=n;i++){
if(k) --k;
j=sa[rk[i]-1];
while(s[i+k]==s[j+k]) ++k;
h[rk[i]]=k;
}
}
inline void prework(){
for(int i=1;i<=n;i++) st[i][0]=h[i];
for(int j=1;(1<<j)<n;j++)
for(int i=1;i+(1<<j)-1<=n;j++)
st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
inline int query(int l,int r){
if(l>r) swap(l,r);++l;
int t=log2(r-l+1); return min(st[l][t],st[r-(1<<t)+1][t]);
}
int main(){
while(~scanf("%d",&k) && k!=0){
scanf("%s",s+1); m=127; n=strlen(s+1);
scanf("%s",a+1); int nn=strlen(a+1);
s[++n]='z'+1;
for(int i=1;i<=nn;i++) s[++n]=a[i]; s[n+1]=0;//!
ssort(); get_h(); prework();
memset(f,0,sizeof f);
LL ans=0,now=0; top=0;
for(int i=1;i<=n;i++){
if(h[i]<k) top=0,now=0;
else{
int tmp=0;
while(stk[top]>h[i] && top){
now+=1LL*(h[i]-stk[top])*f[top];
tmp+=f[top--];
}
if(sa[i-1]>n-nn) tmp++,now+=h[i]-k+1;//sa[i-1]
stk[++top]=h[i]; f[top]=tmp;
if(sa[i]<n-nn) ans+=now;
}
}
top=0,now=0;
for(int i=1;i<=n;i++){
if(h[i]<k) top=0,now=0;
else{
int tmp=0;
while(stk[top]>h[i] && top){
now+=1LL*f[top]*(h[i]-stk[top]);
tmp+=f[top--];
}
if(sa[i-1]<n-nn) tmp++,now+=h[i]-k+1;
stk[++top]=h[i]; f[top]=tmp;
if(sa[i]>n-nn) ans+=now;
}
}
printf("%lld\n",ans);
}
return 0;
}