codeforces954I Yet Another String Matching Problem

题面

题意

定义一种对字符串的操作为将某种字符全部替换为另一种字符,定义两个字符串之间的距离为让两个串相同的最少操作次数.
现给出两个字符串S,T,求S的每个长度为$|$T$|$的子串与T之间的距离.

做法

首先转化一下两个字符串a,b之间的距离,若$a[i]=u,b[i]=v$,则可以由u向v连一条边,这样最后所有联通块的大小减一之和即为答案.
现在考虑如何对所有询问同时快速连边,不难发现这个可以用fft优化,将S串中的所有字符u看作1,其余字符看作0,再将T串中的所有字符v看作1,其余字符看作0.将T串翻转后与S串相乘,即可得到S的哪些子串中的字符u会对应T串中的字符v,然后用并查集维护一下所有连边即可.

代码

#include<bits/stdc++.h>
#define db double
#define N 300100
const db pi=acos(-1);
using namespace std;

int n,m,l,rev[N];
char S[N],T[N];
struct Bcj
{
	int fa[10],ans;
	int ff(int u){return u==fa[u]?u:fa[u]=ff(fa[u]);}
	void init(){int i;for(i=0;i<6;i++) fa[i]=i;}
	void mg(int u,int v)
	{
		u-='a',v-='a';
		u=ff(u),v=ff(v);
		if(u==v) return;
		ans++;
		fa[u]=v;
	}
}bcj[N];
struct Xs
{
	db ss,xs;
	Xs operator + (const Xs &u) const{return (Xs){ss+u.ss,xs+u.xs};}
	Xs operator - (const Xs &u) const{return (Xs){ss-u.ss,xs-u.xs};}
	Xs operator * (const Xs &u) const{return (Xs){ss*u.ss-xs*u.xs,ss*u.xs+xs*u.ss};}
}a[N],b[N];

inline void fft(Xs *a,bool dft)
{
	int i,j,k;
	Xs x,y,dw,now;
	for(i=0;i<l;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
	for(i=1;i<l;i<<=1)
	{
		dw=(Xs){cos(pi/(db)i),sin(pi/(db)i)};
		for(j=0;j<l;j+=(i<<1))
		{
			now=(Xs){1,0};
			for(k=j;k<j+i;k++)
			{
				x=a[k];
				y=a[k+i]*now;
				a[k]=x+y;
				a[k+i]=x-y;
				now=now*dw;
			}
		}
	}
	if(!dft)
	{
		for(i=0;i<l;i++) a[i].ss/=(db)l;
		reverse(a+1,a+l);
	}
}

inline void work(int u,int v)
{
	int i;
	for(i=0;i<l;i++) a[i]=b[i]=(Xs){0,0};
	for(i=0;i<n;i++) a[i].ss=(S[i]==u);
	for(i=0;i<m;i++) b[i].ss=(T[i]==v);
	reverse(b,b+m);
	fft(a,1),fft(b,1);
	for(i=0;i<l;i++) a[i]=a[i]*b[i];
	fft(a,0);
	for(i=0;i<=n-m;i++)
	{
		if(fabs(a[i+m-1].ss)>0.5)
			bcj[i].mg(u,v);
	}
}

int main()
{
	int i,j;
	scanf("%s%s",S,T);
	n=strlen(S);
	m=strlen(T);
	for(i=0;i<=n-m;i++) bcj[i].init();
	for(l=1;l<n;l<<=1);l<<=1;
	for(i=1;i<l;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*l/2);
	for(i='a';i<='f';i++)
	{
		for(j='a';j<='f';j++)
		{
			if(i==j) continue;
			work(i,j);
		}
	}
	for(i=0;i<=n-m;i++) printf("%d ",bcj[i].ans);
}