例题:
https://codeforces.com/gym/101002/attachments
E题 K-inversions
题意:给定一个只含有 B B B和 A A A的字符串,输出其中 B . . . A B...A B...A的数量,按照 A A A到 B B B的距离为 1 , 2... , n 1,2...,n 1,2...,n的形式输出结果
怎么做转换为fft呢,我们要构造两个多项式,其乘起来各项的结果就是对应的 B . . A B..A B..A的数量
假设一对 B . . A B..A B..A A A A到 B B B的距离为 l l l 原串的第 i i i位为 A A A,第 j j j位为 B ( j < i ) B(j<i) B(j<i),那么就会在 l = i − j l=i-j l=i−j的地方产生 1 1 1的贡献
这样我们在原串为
A
A
A的位置对
x
1
[
l
o
a
]
x1[loa]
x1[loa]赋值为
1
1
1,
在原串为
B
B
B的位置对
x
2
[
−
l
o
a
]
x2[-loa]
x2[−loa]赋值为
1
1
1
对于
x
2
x2
x2,为了方便操作,对这个数组下标
+
n
+n
+n
这样答案区间就在
[
l
e
n
,
l
e
n
+
l
e
n
−
1
]
[len,len+len-1]
[len,len+len−1] (下标是从
0
0
0标的,所以起点在
0
+
l
e
n
0+len
0+len)
写的比较标准,用作fft的模板
#include <bits/stdc++.h>
const int maxn=1e7+5;
#define pi acos(-1.0)
using namespace std;
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);
}
};
void change(Complex y[],int len)
{
for(int i=1,j=len/2;i<len-1;i++)
{
if(i<j)swap(y[i],y[j]);
int k=len/2;
while(j>=k)
{
j-=k;
k/=2;
}
if(j<k)
{
j+=k;
}
}
}
void fft(Complex y[],int len,int on)
{
change(y,len);
for(int h=2;h<=len;h<<=1)
{
Complex wn(cos(-on*2*pi/h),sin(-on*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 u=y[k];
Complex t=w*y[k+h/2];
y[k]=u+t;
y[k+h/2]=u-t;
w=w*wn;
}
}
}
if(on==-1)
{
for(int i=0;i<len;i++)
{
y[i].x/=len;
}
}
}
Complex x1[maxn],x2[maxn];
int s1[maxn],s2[maxn];
string s;
int len1,len2;
int main()
{
ios::sync_with_stdio(false);
cin>>s;
len1=(int)s.size();
len2=len1;
for(int i=0;i<len1;i++)
{
if(s[i]=='A')s1[i]=1;
}
for(int i=0;i<len2;i++)
{
if(s[i]=='B')s2[i]=1;
}
reverse(s2,s2+len2);
int len=1;
while(len<len1*2||len<len2*2)
{
len<<=1;
}
for(int i=0;i<len1;i++)
{
x1[i]=Complex(s1[i],0);
}
for(int i=len1;i<len;i++)
{
x1[i]=Complex(0,0);
}
for(int i=0;i<len2;i++)
{
x2[i]=Complex(s2[i],0);
}
for(int i=len2;i<len;i++)
{
x2[i]=Complex(0,0);
}
fft(x1,len,1);
fft(x2,len,1);
for(int i=0;i<len;i++)
{
x1[i]=x1[i]*x2[i];
}
fft(x1,len,-1);
for(int i=len1;i<len1+len2-1;i++)
{
cout<<(int)(x1[i].x+0.5)<<endl;
}
return 0;
}