把A的位置i,对应多项式a的i次的系数为1.
把B的位置i,对应多项式b的n-i-1次的系数为1.
这样a*b的结果得到的多项式c,第n次的系数,就是距离为1的BA。
因为两个多项式相乘后,a次数i,b次数j相乘。
如果i<=j则n-j-1+i <= n-i-1+i=n-1;
如果i>j 则n-j-1+i>n-1;
所以B在A的右侧的对应系数均大于n-1.
当i-j=k时
n-j-1+i=n-1+k;
所以n-1+k次的系数对应的就是BA相距k的个数。
所以我们直接让a*b即可,这个过程是n^2的。
所以要用FFT加速一下,
FFT的过程我也不完全清楚,但是会用就行
先把ab函数,FFT后对应成点位置表示,然后On的求出c的点位置表示
再逆FFT求出c函数
输出就行。由于精度问题,输出时要输出 int(a.re+0.5);
#include<bits/stdc++.h>
using namespace std;
const int MAX=1e6+10;
const int MOD=1e9+7;
const double PI=acos(-1.0);
typedef long long ll;
struct Complex
{
double re,im;
Complex(double _re,double _im){re=_re;im=_im;}
Complex(){}
}a[4*MAX],b[4*MAX];
Complex operator+(const Complex &c1,const Complex &c2){return Complex(c1.re+c2.re,c1.im+c2.im);}
Complex operator-(const Complex &c1,const Complex &c2){return Complex(c1.re-c2.re,c1.im-c2.im);}
Complex operator*(const Complex &c1,const Complex &c2){return Complex(c1.re*c2.re-c1.im*c2.im,c1.re*c2.im+c1.im*c2.re);}
int rev(int x,int len)
{
int ret=0;
for(int i=0;(1<<i)<len;i++)
{
ret<<=1;
if(x&(1<<i))ret|=1;
}
return ret;
}
//DFT==1,FFT,DFT==2,逆FFT
void FFT(Complex *p,int len,int DFT)
{
for(int i=0;i<len;i++)
{
int x=rev(i,len);
if(i<x)swap(p[i],p[x]);
}
for(int i=1;i<len;i*=2)
{
Complex wm=Complex(cos(DFT*PI/i),sin(DFT*PI/i));
for(int k=0;k<len;k+=i*2)
{
Complex w=Complex(1,0);
for(int j=0;j<i;j++)
{
Complex t=w*p[k+j+i];
Complex u=p[k+j];
p[k+j]=u+t;
p[k+j+i]=u-t;
w=w*wm;
}
}
}
if(DFT==-1)for(int i=0;i<len;i++)p[i].re/=len,p[i].im/=len;
}
char s[MAX];
int main()
{
scanf("%s",s);
int n=strlen(s);
int len=0;
//先确定长度,即FFT中分治部分
while((1<<len)<=2*n)len++;
len=1<<len;
for(int i=0;i<n;i++)
if(s[i]=='A')a[i]=Complex(1,0);
else b[n-i-1]=Complex(1,0);
for(int i=n;i<len;i++)a[i]=b[i]=Complex(0,0);
FFT(a,len,1);
FFT(b,len,1);
for(int i=0;i<len;i++)
a[i]=b[i]*a[i];
FFT(a,len,-1);
for(int i=n;i<2*n-1;i++)
printf("%d\n",int(a[i].re+0.5));
return 0;
}