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

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

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

每次某个串匹配到结尾时，B中的某个区间的前缀都会合法，注意到这些合法的前缀长度与A长度相差一定不超过K，于是用一个2*K+1的差分数组记录答案即可。复杂度$O(nlog 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=200010;
int K,m,na,nb,n,d,ans,lg[N],h[N],c[N],x[N],y[N],sa[N],rk[N],st[N][20];
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,0,sizeof(y));
    rep(i,0,m) c[i]=0;
    rep(i,1,n) c[x[i]=s[i]]++;
    rep(i,1,m) c[i]+=c[i-1];
    for (int i=n; i; i--) sa[c[x[i]]--]=i;
    for (int k=1,p=0; p<n; k<<=1,m=p){
        p=0; rep(i,n-k+1,n) y[++p]=i;
        rep(i,1,n) if (sa[i]>k) y[++p]=sa[i]-k;
        rep(i,0,m) c[i]=0;
        rep(i,1,n) c[x[y[i]]]++;
        rep(i,1,m) c[i]+=c[i-1];
        for (int i=n; i; i--) sa[c[x[y[i]]]--]=y[i];
        rep(i,0,n) y[i]=x[i]; p=1; x[sa[1]]=1;
        rep(i,2,n) x[sa[i]]=Cmp(sa[i],sa[i-1],k)?p:++p;
    }
}

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

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

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

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

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+1);
        if (c>=0) col(y-1-c,y-1+c);
        return;
    }
    if (z==K) return;
    dfs(x+1,y,z+1); dfs(x,y+1,z+1); dfs(x+1,y+1,z+1);
}

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