CF954I Yet Another String Matching Problem

Posted troverld

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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

则我们共得到四对需要修改的对:((a-b)(b-c)(a-a)(c-b))

画出图来就是这个:

技术图片

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

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

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

(6^2)枚举这条边的两个端点(例如((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;
} 

以上是关于CF954I Yet Another String Matching Problem的主要内容,如果未能解决你的问题,请参考以下文章

CF-1359 D. Yet Another Yet Another Task ST表+单调队列

CF868F Yet Another Minimization Problem

CF1511C Yet Another Card Deck

CF1511C Yet Another Card Deck

CF1511C Yet Another Card Deck

CF1511C Yet Another Card Deck