[BZOJ4340][BJOI2015]隐身术(后缀数组)

考虑到K很小，于是可以暴搜每次用的是哪种操作，跳过AB相等的字符可以用SA求LCP加速。

主要流程就是，枚举B的每个后缀，对每个后缀统计合法前缀个数。DFS搜索每次决策，用SA跳过相同字符，当A或B匹配到结尾时统计答案。

每次某个串匹配到结尾时，B中的某个区间的前缀都会合法，注意到这些合法的前缀长度与A长度相差一定不超过K，于是用一个2*K+1的差分数组记录答案即可。复杂度$O(n\log n+n3^K)$

 #include<cstring>
 #include<cstdio>
 #include<algorithm>
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 using namespace std;

 const int N=;
 int K,m,na,nb,n,d,ans,lg[N],h[N],c[N],x[N],y[N],sa[N],rk[N],st[N][];
 char s[N],A[N],B[N];

 bool Cmp(int a,int b,int l){ return y[a]==y[b] && y[a+l]==y[b+l]; }

 void getSA(int m){
     memset(y,,sizeof(y));
     rep(i,,m) c[i]=;
     rep(i,,n) c[x[i]=s[i]]++;
     rep(i,,m) c[i]+=c[i-];
     for (int i=n; i; i--) sa[c[x[i]]--]=i;
     for (int k=,p=; p<n; k<<=,m=p){
         p=; rep(i,n-k+,n) y[++p]=i;
         rep(i,,n) if (sa[i]>k) y[++p]=sa[i]-k;
         rep(i,,m) c[i]=;
         rep(i,,n) c[x[y[i]]]++;
         rep(i,,m) c[i]+=c[i-];
         for (int i=n; i; i--) sa[c[x[y[i]]]--]=y[i];
         rep(i,,n) y[i]=x[i]; p=; x[sa[]]=;
         rep(i,,n) x[sa[i]]=Cmp(sa[i],sa[i-],k)?p:++p;
     }
 }

 void getH(){
     rep(i,,n) rk[sa[i]]=i; int k=;
     rep(i,,n){
         for (int j=sa[rk[i]-]; j+k<=n && i+k<=n && s[i+k]==s[j+k]; k++);
         h[rk[i]]=k; if (k) k--;
     }
 }

 void init(){
     lg[]=; rep(i,,n) lg[i]=lg[i>>]+;
     rep(i,,n) st[i][]=h[i];
     rep(i,,) rep(j,,n-(<<i)+) st[j][i]=min(st[j][i-],st[j+(<<(i-))][i-]);
 }

 int lcp(int l,int r){
     int x=rk[l],y=rk[r+na+];
     if (x>y) swap(x,y);
     x++; int t=lg[y-x+];
     return min(st[x][t],st[y-(<<t)+][t]);
 }

 void col(int l,int r){ l=max(l,d); r=min(r,nb); c[K-(na-(l-d+))+]++; c[K-(na-(r-d+))+]--; }

 void dfs(int x,int y,int z){
     int t=lcp(x,y); x+=t; y+=t;
     if (x>na || y>nb){
         int c=K-z-(na-x+);
         if (c>=) col(y--c,y-+c);
         return;
     }
     if (z==K) return;
     dfs(x+,y,z+); dfs(x,y+,z+); dfs(x+,y+,z+);
 }

 int main(){
     freopen("bzoj4340.in","r",stdin);
     freopen("bzoj4340.out","w",stdout);
     scanf("%d%s%s",&K,A+,B+); m=*K+;
     na=strlen(A+); nb=strlen(B+);
     rep(i,,na) s[++n]=A[i]; s[++n]='#';
     rep(i,,nb) s[++n]=B[i];
     getSA(); getH(); init();
     for (d=; d<=nb; d++){
         rep(j,,m) c[j]=;
         dfs(,d,);
         rep(j,,m){
             c[j]+=c[j-];
             if (c[j]) ans++;
         }
     }
     printf("%d\n",ans);
     return ;
 }