“21 天好习惯”第一期－1

CF528D Fuzzy Search

传送门

$k=0$时

题目保证只有四个字母$"ATGC"$，暗示我们可以先分成4种情况处理，最后加起来。

比如说文本串$S="TAATGCA"$和模板串$T="AATGC"$

先处理$A$的情况：

$S="\#AA\#\#\#A"$

$T="AA\#\#\#"$

以$S(5)$（下标以0开始）结尾的长度为5的子串和模板串匹配的长度为2

同理，处理$T$的情况，以$S(5)$结尾的长度为5的子串和模板串匹配的长度为1

处理$C$的情况，以$S(5)$结尾的长度为5的子串和模板串匹配的长度为1

处理$G$的情况，以$S(5)$结尾的长度为5的子串和模板串匹配的长度为1

加起来后可知，以$S(5)$结尾的长度为5的子串和模板串匹配的长度为5，即以$S(5)$结尾的长度为5的子串和模板串完全匹配

$k>0$时，可以理解为将文本串中的字母$C$（当前在处理字母$c$的情况）往两边扩散$k$个距离。

比如$k=1$，此时先处理$A$的情况：

$S="AAAA\#\#A"$

$T="AA\#\#\#"$

以$S(5)$（下标以0开始）结尾的长度为5的子串和模板串匹配的长度为2

当前处理字符$c$时，接下来就是构造多项式$f={\sum }_{i=0}^{n-1}[S[i]={=}^{\prime }{c}^{\prime }]\ast x^i$，多项式$g={\sum }_{i=0}^{m-1}[T[m-1-i]={=}^{\prime }{c}^{\prime }]\ast x^i$，然后$a=f\ast s$，$a(d)={\sum }_{i+j=d}[S[i]={=}^{\prime }{c}^{\prime }\&\&T[m-1-j]={=}^{\prime }{c}^{\prime }]x^d$就表示以$S(i)$结尾的长度为$m-1$的子串和模板串匹配的长度。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=(1<<19)+7,mod=998244353;
inline int qpow(int x,int y) {
	int res(1);
	while(y) {
		if(y&1) res=1LL*res*x%mod;
		x=1LL*x*x%mod;
		y>>=1;
	}
	return res;
}
int r[maxn];
inline void ntt(int *x,int lim,int opt) {
	int i,j,k,m,gn,g,tmp;
	for(i=0; i<lim; ++i)
		if(i<r[i]) swap(x[i],x[r[i]]);
	for(m=2; m<=lim; m<<=1) {
		k=m>>1;
		gn=qpow(3,(mod-1)/m);
		for(i=0; i<lim; i+=m) {
			g=1;
			for(j=0; j<k; ++j,g=1LL*g*gn%mod) {
				tmp=1LL*x[i+j+k]*g%mod;
				x[i+j+k]=(x[i+j]-tmp+mod)%mod;
				x[i+j]=(x[i+j]+tmp)%mod;
			}
		}
	}
	if(opt==-1) {
		reverse(x+1,x+lim);
		int inv=qpow(lim,mod-2);
		for(i=0; i<lim; ++i) x[i]=1LL*x[i]*inv%mod;
	}
}
int A[maxn],B[maxn],cnt[maxn],n,m,k,lim(1);
char a[maxn],b[maxn];
void solve(char c) {
	for(int i=0;i<lim;++i) A[i]=B[i]=0;
	for(int i=0,last=-1e9;i<n;++i) {
		if(a[i]==c) last=i;
		if(i-last<=k) A[i]=1;
	}
	for(int i=n-1,last=1e9;~i;--i) {
		if(a[i]==c) last=i;
		if(last-i<=k) A[i]=1;
	}
	for(int i=0;i<m;++i) B[i]=(b[m-1-i]==c);
	ntt(A,lim,1),ntt(B,lim,1);
	for(int i=0; i<lim; ++i) A[i]=1LL*A[i]*B[i]%mod;
	ntt(A,lim,-1);
	for(int i=0;i<lim;++i) cnt[i]+=A[i];
}
int main() {
	scanf("%d%d%d",&n,&m,&k);
	scanf("%s%s",a,b);
	while(lim<(n<<1)) lim<<=1;
	for(int i=0; i<lim; ++i) r[i]=(i&1)*(lim>>1)+(r[i>>1] >> 1);
	for(int i=0;i<4;++i) solve("ATGC"[i]);
	int ans(0);
	for(int i=m-1;i<lim;++i) ans+=(cnt[i]==m);
	cout<<ans<<'\n';
	return 0;
}

P4173 残缺的字符串

传送门

文本串为$A(长度为n)$，模式串为$B(长度为m)$

不考虑通配符的时候，可以用完全匹配函数可以是$P(x)={\sum }_{i=0}^{m-1}[B(i)-A(x-m+1+i){]}^2$

将通配符赋值为$0$，那么有完全匹配函数$P(x)={\sum }_{i=0}^{m-1}[B(i)-A(x-m+1+i){]}^2\ast A(x-m+1+i)\ast B(i)$

老套路：$S(i)=B(m-1-i),B(i)=S(m-1-i)$

那么有完全匹配函数$P(x)={\sum }_{i=0}^{m-1}S(m-1-i{)}^3\ast A(x-m+1+i)+{\sum }_{i=0}^{m-1}A(x-m+1+i{)}^3\ast S(m-1-i)-2\ast {\sum }_{i=0}^{m-1}A(x-m+1+i{)}^2\ast S(m-1-i{)}^2$

$P(x)={\sum }_{i+j=x}S(i{)}^3\ast A(j)+{\sum }_{i+j=x}A(i{)}^3\ast S(j)-2\ast {\sum }_{i+j=x}A(i{)}^2\ast S(j{)}^2$

做7次$FFT$就好了，浮点数精度可能会有问题，$p(x)<1$就可以看作$0$了，我看有的人是$p(x)<1e-7$，而我改成$p(x)<1e-6$都wa。

用$NTT$涉及取余（$p(x)$会大于$mod(998244353)$），可能会影响结果？我试了一下只有65分。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=(1<<20)+7;
const double PI=acos(-1.0);
struct Complex{
	double x,y;
	Complex(double _x=0.0,double _y=0.0) {
		x=_x;
		y=_y;
	}
	Complex operator-(const Complex &b) const {
		return Complex(x-b.x,y-b.y);
	}
	Complex operator+(const Complex &b) const {
		return Complex(x+b.x,y+b.y);
	}
	Complex operator*(const Complex &b) const {
		return Complex(x*b.x-y*b.y,x*b.y+y*b.x);
	}
};
int r[maxn];
inline void fft(Complex *y,int len,int opt) {
	for(int i=0;i<len;++i)
		if(i<r[i]) swap(y[i],y[r[i]]);
	for(int h=2;h<=len;h<<=1) {
		Complex wn(cos(2*PI/h),sin(opt*2*PI/h));
		for(int j=0;j<len;j+=h) {
			Complex w(1,0);
			for(int k=j;k<j+h/2;++k) {
				Complex tmp=w*y[k+h/2];
				y[k+h/2]=y[k]-tmp;
				y[k]=y[k]+tmp;
				w=w*wn;
			}
		}
	}
	if(opt==-1) {
		for(int i=0;i<len;++i) y[i].x/=len;
	}
}
Complex A[maxn],B[maxn],p[maxn];
double x[maxn],y[maxn];
char a[maxn/2],b[maxn/2];
int main() {
	int n,m,k,lim(1);
	scanf("%d%d",&m,&n);
	scanf("%s%s",b,a);
	while(lim<(n<<1)) lim<<=1;
	for(int i=0; i<lim; ++i) r[i]=(i&1)*(lim>>1)+(r[i>>1] >> 1);
	for(int i=0;i<n;++i) x[i]=(a[i]!='*')?a[i]-'a'+1:0,A[i]=Complex(x[i]*x[i]*x[i],0);
	for(int i=0;i<m;++i) y[i]=(b[m-1-i]!='*')?b[m-1-i]-'a'+1:0,B[i]=Complex(y[i],0);
	fft(A,lim,1),fft(B,lim,1);
	for(int i=0;i<lim;++i) {
		p[i]=p[i]+A[i]*B[i];
		A[i]=Complex(x[i],0);
		B[i]=Complex(y[i]*y[i]*y[i],0);
	}
	fft(A,lim,1),fft(B,lim,1);
	for(int i=0;i<lim;++i) {
		p[i]=p[i]+A[i]*B[i];
		A[i]=Complex(x[i]*x[i],0);
		B[i]=Complex(y[i]*y[i],0);
	}
	fft(A,lim,1),fft(B,lim,1);
	for(int i=0;i<lim;++i) p[i]=p[i]-A[i]*B[i]*Complex(2,0);
	fft(p,lim,-1);
	vector<int>ans;
	for(int i=m-1;i<n;++i) {
		if(fabs(p[i].x)<1) ans.emplace_back(i-m+2);
	}
	cout<<ans.size()<<'\n';
	for(int i:ans) cout<<i<<' ';
	return 0;
}