P4173 残缺的字符串

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

不考虑通配符的时候,可以用完全匹配函数可以是P(x)=i=0m1[B(i)A(xm+1+i)<mtext> </mtext>]2{P(x)=\sum_{i=0}^{m-1}{[B(i)-A(x-m+1+i)\ ]^2}}P(x)=i=0m1[B(i)A(xm+1+i) ]2

将通配符赋值为000,那么有完全匹配函数P(x)=i=0m1[B(i)A(xm+1+i)<mtext> </mtext>]2A(xm+1+i)B(i){P(x)=\sum_{i=0}^{m-1}{[B(i)-A(x-m+1+i)\ ]^2*A(x-m+1+i)*B(i)}}P(x)=i=0m1[B(i)A(xm+1+i) ]2A(xm+1+i)B(i)

老套路:S(i)=B(m1i),B(i)=S(m1i)S(i)=B(m-1-i),B(i)=S(m-1-i)S(i)=B(m1i),B(i)=S(m1i)

那么有完全匹配函数P(x)=i=0m1S(m1i)3A(xm+1+i)+i=0m1A(xm+1+i)3S(m1i)2i=0m1A(xm+1+i)2S(m1i)2{P(x)=\sum_{i=0}^{m-1}{S(m-1-i)^3*A(x-m+1+i)}+\sum_{i=0}^{m-1}{A(x-m+1+i)^3*S(m-1-i)}-2*\sum_{i=0}^{m-1}{A(x-m+1+i)^2*S(m-1-i)^2}}P(x)=i=0m1S(m1i)3A(xm+1+i)+i=0m1A(xm+1+i)3S(m1i)2i=0m1A(xm+1+i)2S(m1i)2

P(x)=i+j=xS(i)3A(j)+i+j=xA(i)3S(j)2i+j=xA(i)2S(j)2{P(x)=\sum_{i+j=x}{S(i)^3*A(j)}+\sum_{i+j=x}{A(i)^3*S(j)}-2*\sum_{i+j=x}{A(i)^2*S(j)^2}}P(x)=i+j=xS(i)3A(j)+i+j=xA(i)3S(j)2i+j=xA(i)2S(j)2

做7次FFTFFTFFT就好了,浮点数精度可能会有问题,p(x)<1p(x)<1p(x)<1就可以看作000了,我看有的人是p(x)<1e7p(x)<1e-7p(x)<1e7,而我改成p(x)<1e6p(x)<1e-6p(x)<1e6都wa。

NTTNTTNTT涉及取余(p(x)p(x)p(x)会大于mod(998244353)mod(998244353)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;
}
FFT 文章被收录于专栏

1

全部评论

相关推荐

上周组里招人,我面了六个候选人,回来跟同事吃饭的时候聊起一个让我挺感慨的现象。前三个候选人,算法题写得都不错。第一道二分查找,五分钟之内给出解法,边界条件也处理得干净。第二道动态规划,状态转移方程写对了,空间复杂度也优化了一版。我翻他们的简历,力扣刷题量都在300以上。后三个呢,就有点参差不齐了。有的边界条件没处理好,有的直接说这道题没刷过能不能换个思路讲讲。其中有一个女生,我印象特别深——她拿到题之后没有马上写,而是先问我:“面试官,我能先跟你确认一下我对题目的理解吗?”然后她把自己的思路讲了一遍,虽然最后代码写得不是最优解,但整个沟通过程非常顺畅。这个女生的代码不是最优的,但当我问她“如果这里是线上环境,你会怎么设计’的时候,她给我讲了一套完整的方案——异常怎么处理、日志怎么打、怎么平滑发布。她对这是之前在实习的时候踩过的坑。”我在想LeetCode到底在筛选什么?我自己的经历可能有点代表性。我当年校招的时候,也是刷了三百多道题才敢去面试。那时候大家都刷,你不刷就过不了笔试关。后来工作了,前三年基本没再打开过力扣。真正干活的时候,没人让你写反转链表,也没人让你手撕红黑树。更多的是:这个接口为什么慢了、那个服务为什么OOM了、线上数据对不上了得排查一下。所以后来我当面试官,慢慢调整了自己的评判标准。算法题我还会出,但目的变了。我出算法题,不是想看你能不能背出最优解。而是想看你拿到一个陌生问题的时候,是怎么思考的。你会先理清题意吗?你会主动问边界条件吗?你想不出来的时候会怎么办?你写出来的代码,变量命名乱不乱、结构清不清楚?这些才是工作中真正用得到的能力。LeetCode是一个工具,不是目的。它帮你熟悉数据结构和常见算法思路,这没问题。但如果你刷了三百道题,却说不清楚自己的项目解决了什么问题、遇到了什么困难、你是怎么解决的,那这三百道题可能真的白刷了。所以还要不要刷LeetCode?要刷,但别只刷题。刷题的时候,多问自己几个为什么:为什么用这个数据结构?为什么这个解法比那个好?如果换个条件,解法还成立吗?把刷题当成锻炼思维的方式,而不是背答案的任务。毕竟面试官想看到的,从来不是一台背题机器,而是一个能解决问题的人。
国企上岸了的向宇同桌...:最害怕答非所问了,但是频繁反问确定意思又害怕面试官觉得我笨
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务