The North American Invitational Programming Contest 2016： E - K-Inversions Gym - 101002E

 

把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;
}