4836: [Lydsy2017年4月月赛]二元运算

4836: [Lydsy2017年4月月赛]二元运算

Time Limit: 8 Sec Memory Limit: 128 MB

Submit: 286 Solved: 92

[Submit][Status][Discuss]

Description

定义二元运算 opt 满足



现在给定一个长为 n 的数列 a 和一个长为 m 的数列 b ，接下来有 q 次询问。每次询问给定一个数字 c

你需要求出有多少对 (i, j) 使得 a_i opt b_j=c 。

Input

第一行是一个整数 T (1≤T≤10) ，表示测试数据的组数。

对于每组测试数据：

第一行是三个整数 n,m,q (1≤n,m,q≤50000) 。

第二行是 n 个整数，表示 a_1,a_2,?,a_n (0≤a_1,a_2,?,a_n≤50000) 。

第三行是 m 个整数，表示 b_1,b_2,?,b_m (0≤b_1,b_2,?,b_m≤50000) 。

第四行是 q 个整数，第 i 个整数 c_i (0≤c_i≤100000) 表示第 i 次查询的数。

Output

对于每次查询，输出一行，包含一个整数，表示满足条件的 (i, j) 对的个数。

Sample Input

2

2 1 5

1 3

2

1 2 3 4 5

2 2 5

1 3

2 4

1 2 3 4 5

Sample Output

1

0

1

0

0

1

0

1

0

1

HINT

Source

鸣谢Tangjz提供试题

[Submit][Status][Discuss]

先考虑ai≥bj的情形

考虑到每个数字不超过50000，记c=50000,b′i=c−bi

那么，ai+b′j=c+ai−b′j

定义Ai为数组a中数字i出现的次数

然后让A和B′做卷积，其中第i项系数就是差值为i−c的数量了

所以有用的系数都是大于等于50000的项

这一部分的复杂度为O(klogk)

不过对于ai<bj的情形就不能这样了

考虑对值域分治然后做卷积

每次在A中填入[l,mid]在B中填入[mid+1,r]

这样做卷积就能避开所有非法情形了

这一部分复杂度O(klog2k)

很科学对不对，很科学对不对？？？



我也不知道为什么。。。。反正真的卡了好久好久的常数

以及一开始的几个TLE，其实答案也是错的。。不过根本跑不完

首先，这道题卡什么都卡得丧心病狂

模数用1004535809是根本不够的

于是百度找到了这个3221225473，原根是5

存储得用unsigned int平方得用unsigned long long

然后开始解决漫长的卡常数问题。。。

千万不要在NTT过程中写很多除法，klog2k次不是开玩笑的

传指针可能有些慢？所以分开来写两种NTT

随意试了一下发现，让初始值域为[0,65535]使得每一层值域大小都为2k能够显著提升效率

用三目运算符配合加减法加速加法和减法的取模

于是乎终于卡过了。。。(不过还是非常非常慢的)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

const int maxn = (1 << 17);
typedef long long LL;
typedef unsigned int u32;
typedef unsigned long long u64;
const u32 W = 1;
const u32 TT = 10;
const u64 WW = 1;
const u32 mo = 3221225473LL;

int n,m,q,T,A[maxn],B[maxn],ca[maxn],cb[maxn],f[maxn],g[maxn],LOG[maxn + 1];
u32 Ans[maxn],w[maxn + 1],_w[maxn + 1],Inv[maxn + 233],a[maxn],b[maxn],c[maxn];

#define Mul(x,y) (WW * (x) * (y) % mo)
#define max(a,b) ((a) > (b) ? (a) : (b))
#define swap(x,y) ((x) ^= (y),(y) ^= (x),(x) ^= (y))
#define Dec(x,y) ((x) < (y) ? (x) + mo - (y) : (x) - (y))
#define Add(x,y) (mo - (x) > (y) ? (x) + (y) : (x) - mo + (y))

inline u32 ksm(u32 x,u32 y)
{
u32 ret = W;
for (; y; y >>= W)
{
if (y & W) ret = Mul(ret,x);
x = Mul(x,x);
}
return ret;
}

inline int getint()
{
char ch = getchar(); int ret = 0;
while (ch < '0' || '9' < ch) ch = getchar();
while ('0' <= ch && ch <= '9')
ret = ret * 10 + ch - '0',ch = getchar();
return ret;
}

char s[20];
inline void Print(u32 x)
{
if (!x) {puts("0"); return;} int len = 0;
while (x) s[++len] = x % TT,x /= TT;
for (int i = len; i; i--) putchar(s[i] + '0'); puts("");
}

inline void Rader(u32 *F,int N)
{
int j = (N >> 1);
for (int i = 1; i < N - 1; i++)
{
if (i < j) swap(F[i],F[j]); int k = (N >> 1);
while (j >= k) j -= k,k >>= 1; j += k;
}
}

inline void NTT(u32 *F,int N)
{
Rader(F,N);
for (int k = 2; k <= N; k <<= 1)
{
int tmp = k >> 1,G = maxn / k;
for (int i = 0; i < N; i += k)
{
int now = 0;
for (int j = i; j < i + tmp; j++)
{
u32 u = F[j],v = Mul(w[now],F[j + tmp]);
F[j] = Add(u,v); F[j + tmp] = Dec(u,v); now += G;
}
}
}
}

inline void _NTT(u32 *F,int N)
{
Rader(F,N);
for (int k = 2; k <= N; k <<= 1)
{
int tmp = k >> 1,G = maxn / k;
for (int i = 0; i < N; i += k)
{
int now = 0;
for (int j = i; j < i + tmp; j++)
{
u32 u = F[j],v = Mul(_w[now],F[j + tmp]);
F[j] = Add(u,v); F[j + tmp] = Dec(u,v); now += G;
}
}
}
for (int i = 0; i < N; i++) F[i] = Mul(F[i],Inv
);
}

inline void Work()
{
int N = 131072,t = 65536;
for (int i = 1; i <= n; i++) ++a[A[i]];
for (int i = 1; i <= m; i++) ++b[t - B[i]];
NTT(a,N); NTT(b,N);
for (int i = 0; i < N; i++) c[i] = Mul(a[i],b[i]); _NTT(c,N);
for (int i = t; i < N; i++) Ans[i - t] += c[i];
for (int i = 0; i < N; i++) a[i] = b[i] = c[i] = 0;
}

inline void Calc(int l,int r)
{
if (l == r) return;
int mid = l + r >> 1,M = r - l + 1 << 1,tf = 0,tg = 0;
for (int i = l; i <= mid; i++)
if (ca[i]) a[i - l] = ca[i],f[++tf] = i;
for (int i = mid + 1; i <= r; i++)
if (cb[i]) b[i - l] = cb[i],g[++tg] = i;
if (1LL * tf * tg > 1LL * M * LOG[M])
{
NTT(a,M); NTT(b,M);
for (int i = 0; i < M; i++) c[i] = Mul(a[i],b[i]); _NTT(c,M);
for (int i = 0; i < M; i++) Ans[i + l * 2] += c[i],a[i] = b[i] = c[i] = 0;
}
else
{
for (int i = 1; i <= tf; i++)
for (int j = 1; j <= tg; j++)
Ans[f[i] + g[j]] += W * ca[f[i]] * cb[g[j]];
for (int i = 1; i <= tf; i++) a[f[i] - l] = 0;
for (int i = 1; i <= tg; i++) b[g[i] - l] = 0;
}
Calc(l,mid); Calc(mid + 1,r);
}

inline void Solve()
{
n = getint(); m = getint(); q = getint();
for (int i = 1; i <= n; i++) A[i] = getint(),++ca[A[i]];
for (int i = 1; i <= m; i++) B[i] = getint(),++cb[B[i]];
Work(); Calc(0,65535); while (q--) Print(Ans[getint()]);
}

inline void Clear()
{
memset(ca,0,sizeof(ca));
memset(cb,0,sizeof(cb));
memset(Ans,0,sizeof(Ans));
}

int main()
{
#ifdef DMC
freopen("binop1.in","r",stdin);
freopen("test.out","w",stdout);
#endif

w[0] = 1; w[1] = ksm(5,(mo - 1) / maxn);
for (int i = 2; i <= maxn; i++) w[i] = Mul(w[i - 1],w[1]);
for (int i = 0; i <= maxn; i++) _w[i] = w[maxn - i];
for (int i = 1; i <= maxn; i <<= 1) Inv[i] = ksm(i,mo - 2);
for (int i = 2; i <= maxn; i <<= 1) LOG[i] = LOG[i >> 1] + 1;
T = getint(); while (T--) Solve(),Clear();
return 0;
}