题面
题目定义了一种奇怪的运算,是这样的
还给出了两个数组A,B,数组长度和元素大小都<=5*10^4。
有多个询问,给出一个k,问有多少对(i,j)使得
Ai opt Bj = k。
如果 x opt y = x+y,那么显然是一个桶状数组的卷积,FFT可以解决
若 x opt y = x-y,根据FFT题的套路,把一个数组反过来,发现它依然是个卷积。而题目把运算按x与y的大小关系分成了两种,两种都很类似,可以独立地用相同的方法做。
这题是一个不熟FFT的大佬搜CDQ分治搜到的,让我非常感兴趣,所以就来考虑分治。
其实也很简单,不过思路我没听过而已。
对于权值区间[l,r],我们二分一个mid,然后用FFT将[l,mid]和[mid+1,r]做个卷积,翻转一边,再做个卷积,统计答案。正确地清零,保证复杂度。再递归处理两个区间,然后就搞定了。
时间复杂度 O(N (log N)^2)
N=5*10^4。
为什么NTT在电脑上跑得飞快,交上去老是T。
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>
using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))
typedef long long LL;
void read(int &hy)
{
hy=0;
char cc=getchar();
while(cc>'9'||cc<'0')
cc=getchar();
while(cc>='0'&&cc<='9')
{
hy=(hy<<3)+(hy<<1)+cc-