CF954I Yet Another String Matching Problem

VII.CF954I Yet Another String Matching Problem

FFT做字符串匹配就是有意思

我们首先必须要搞清楚暴力匹配是什么样的过程。

例:

位置 0 1 2 3
t的子串 a b a c
s b c a b

则我们共得到四对需要修改的对:(ab)(bc)(aa)(cb)

画出图来就是这个:

如果想一想就可以发现,我们在每一块连通块中求出一棵生成树,则生成树的边数就是处理完这个连通块所需要的次数。显然,该次数为(连通块大小-1)。

我们可以采用冰茶姬。对于每一条新边,在冰茶姬中连边,如果连边成功,答案加1;否则,即两个点已经在同一个连通块中,忽略不计。

如何求出对于t的每个子串,它与s间到底连了哪些边呢?我们采用FFT。

62枚举这条边的两个端点(例如(a,b)(b,a)(a,f)),然后直接FFT一卷完事,同我们前几道题的方法完全一致。

代码:

#include<bits/stdc++.h>
using namespace std;
const double pi=acos(-1);
int lim=1,lg,rev[1<<20],dsu[125010][6],ans[125010],S,T;
char s[125010],t[125010];
int ask(int x,int y){
	return dsu[x][y]==y?y:dsu[x][y]=ask(x,dsu[x][y]);
}
bool merge(int x,int y,int z){
	y=ask(x,y),z=ask(x,z);
	if(y==z)return false;
	dsu[x][y]=dsu[x][z];
	return true;
}
struct cp{
	double x,y;
	cp(double u=0,double v=0){x=u,y=v;}
	friend cp operator +(const cp &u,const cp &v){return cp(u.x+v.x,u.y+v.y);}
	friend cp operator -(const cp &u,const cp &v){return cp(u.x-v.x,u.y-v.y);}
	friend cp operator *(const cp &u,const cp &v){return cp(u.x*v.x-u.y*v.y,u.x*v.y+u.y*v.x);}
}f[1<<20],g[1<<20];
void FFT(cp *a,int tp){
	for(int i=0;i<lim;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
	for(int md=1;md<lim;md<<=1){
		cp rt=cp(cos(pi/md),tp*sin(pi/md));
		for(int stp=md<<1,pos=0;pos<lim;pos+=stp){
			cp w=cp(1,0);
			for(int i=0;i<md;i++,w=w*rt){
				cp x=a[pos+i],y=w*a[pos+md+i];
				a[pos+i]=x+y;
				a[pos+md+i]=x-y;
			}
		}
	}
}
void match(int x,int y){
	for(int i=0;i<lim;i++)f[i]=g[i]=cp(0,0);
	for(int i=0;i<S;i++)f[i]=cp(s[i]-'a'==x,0);
	for(int i=0;i<T;i++)g[i]=cp(t[i]-'a'==y,0);
	FFT(f,1),FFT(g,1);
	for(int i=0;i<lim;i++)f[i]=f[i]*g[i];
	FFT(f,-1);
	for(int i=S-1;i<T;i++)if((int)(f[i].x/lim+0.5)>=1)ans[i]+=merge(i,x,y);
}
int main(){
	scanf("%s%s",t,s),T=strlen(t),S=strlen(s),reverse(s,s+S);
	while(lim<=(S+T))lim<<=1,lg++;
	for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
	for(int i=S-1;i<T;i++)for(int j=0;j<6;j++)dsu[i][j]=j;
	for(int i=0;i<6;i++)for(int j=0;j<6;j++)match(i,j);
	for(int i=S-1;i<T;i++)printf("%d ",ans[i]);
	return 0;
} 
posted @   Troverld  阅读(147)  评论(0编辑  收藏  举报
编辑推荐:
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 【杂谈】分布式事务——高大上的无用知识?
点击右上角即可分享
微信分享提示