题目:
题意:
给定长度为N的只包含A和B的字符串,定义BA为逆序对,如果它们的距离为K,则称为:K-Inversions,求出K-Inversions的数量,对于K属于【1,N-1】
分析:
刚学完神奇的FFT,但还是有诸多疑惑,所以这里只记录个板子
对于这道题要怎么转换为FFT呢?设A的位置为x,B的位置为y,则:x - y = k;如果将位置看成是多项式的幂,就可以在O(NlogN)内求出所有幂为x+y的系数,所以就将y = (N-y-1),那么x - y = x - N + y + 1 = k,即:x + y = N + k - 1,所以只要B的位置反转一下,就可以套用FFT完美的解决了
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 3e6+10;
const double Pi = acos(-1.0);
struct Complex {
double x,y;
Complex(double xx = 0, double yy = 0) {x = xx,y = yy;}
}a[maxn],b[maxn];
Complex