【BZOJ4892】【TJOI2017】—DNA（后缀数组+ST表）

传送门

考虑3个不同的位置

就是枚举$A$的一个子串，看和$B$是否有超过3个位置不匹配

可以建立后缀数组之后跳$lcp$,如果有$3$个位置以上不匹配就不行

似乎直接哈希二分就完了啊
写个p的后缀数组

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	char ch=getchar();
	int res=0,f=1;
	while(!isdigit(ch))f^=(ch=='-'),ch=getchar();
	while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
	return f?res:-res;
}
const int N=200005;
int lena,lenb,ans;
char s[N];
namespace Sa{
	int n,m,sa[N],sa2[N],cnt[N],rk[N],ht[N],st[22][N<<1],lg[N];
	inline void bucket_sort(){
		for(int i=1;i<=m;i++)cnt[i]=0;
		for(int i=1;i<=n;i++)cnt[rk[sa2[i]]]++;
		for(int i=1;i<=m;i++)cnt[i]+=cnt[i-1];
		for(int i=n;i>=1;i--)sa[cnt[rk[sa2[i]]]--]=sa2[i];
	}
	inline void buildsa(){
		m=128;
		for(int i=1;i<=n;i++)rk[i]=s[i],sa2[i]=i;
		bucket_sort();
		for(int i=1,pos=0;i<=n&&pos<n;i<<=1){
			pos=0;
			for(int j=n-i+1;j<=n;j++)sa2[++pos]=j;
			for(int j=1;j<=n;j++)if(sa[j]>i)sa2[++pos]=sa[j]-i;
			bucket_sort();
			swap(rk,sa2);
			rk[sa[1]]=1,pos=1;
			for(int j=2;j<=n;j++)
				rk[sa[j]]=((sa2[sa[j]]==sa2[sa[j-1]])&&(sa2[sa[j]+i]==sa2[sa[j-1]+i])?pos:++pos);
			m=pos;
		}
		for(int i=1;i<=n;i++)rk[sa[i]]=i;
		for(int i=1,k=0,j;i<=n;ht[rk[i++]]=k){
			for(k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
		}
	}
	inline void buildst(){
		lg[0]=-1;
		for(int i=1;i<=n;i++)st[0][i]=ht[i],lg[i]=lg[i>>1]+1;
		for(int i=1;(1<<i)<=n;i++)
			for(int j=1;j+(1<<i)-1<=n;j++)
				st[i][j]=min(st[i-1][j],st[i-1][j+(1<<(i-1))]);
	}
	inline int lcp(int x,int y){
		x=rk[x],y=rk[y];
		if(x>y)swap(x,y);x++;
		int t=lg[y-x+1];
		return min(st[t][x],st[t][y-(1<<t)+1]);
	}
}
using namespace Sa;
int main(){
	int T=read();
	while(T--){
		ans=0;
		scanf("%s",s+1);
		lena=strlen(s+1);
		s[lena+1]='$';
		scanf("%s",s+lena+2);
		lenb=strlen(s+lena+2);
		n=lena+lenb+1;
		buildsa(),buildst();
		for(int i=1;i<=lena-lenb+1;i++){
			int num=0;
			for(int j=1;j<=lenb&&num<=3;){
				if(s[i+j-1]==s[lena+j+1]){
					j+=lcp(i+j-1,lena+j+1);
				}
				else num++,j++;	
			}if(num<=3)ans++;
		}
		cout<<ans<<'
';
	}
}