BZOJ4836: [Lydsy1704月赛]二元运算【分治FFT】【卡常(没卡过)】
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
思路
在上面两种情况中,减法可以把y取反直接fft,然后加法因为有限制可以在值域上进行分治fft
然后其实很板子的。。。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
namespace io {
const int BUFSIZE = 1 << 20;
char ibuf[BUFSIZE], *is = ibuf, *it = ibuf;
char obuf[BUFSIZE], *os = obuf, *ot = obuf + BUFSIZE - 1;
char read_char() {
if (is == it)
it = (is = ibuf) + fread(ibuf, 1, BUFSIZE, stdin);
return *is++;
}
int read_int() {
int x = 0, f = 1;
char c = read_char();
while (!isdigit(c)) {
if (c == '-') f = -1;
c = read_char();
}
while (isdigit(c)) x = x * 10 + c - '0', c = read_char();
return x * f;
}
ll read_ll() {
ll x = 0, f = 1;
char c = read_char();
while (!isdigit(c)) {
if (c == '-') f = -1;
c = read_char();
}
while (isdigit(c)) x = x * 10 + c - '0', c = read_char();
return x * f;
}
void read_string(char* s) {
char c = read_char();
while (isspace(c)) c = read_char();
while (!isspace(c)) *s++ = c, c = read_char();
*s = 0;
}
void flush() {
fwrite(obuf, 1, os - obuf, stdout);
os = obuf;
}
void print_char(char c) {
*os++ = c;
if (os == ot) flush();
}
void print_int(int x) {
static char q[20];
if (!x) print_char('0');
else {
if (x < 0) print_char('-'), x = -x;
int top = 0;
while (x) q[top++] = x % 10 + '0', x /= 10;
while (top--) print_char(q[top]);
}
}
void print_ll(ll x) {
static char q[20];
if (!x) print_char('0');
else {
if (x < 0) print_char('-'), x = -x;
int top = 0;
while (x) q[top++] = x % 10 + '0', x /= 10;
while (top--) print_char(q[top]);
}
}
struct flusher_t {
~flusher_t() {
flush();
}
} flusher;
};
using namespace io;
const int N = 3e5 + 10;
const int M = 5e4 + 10;
const double eps = 1e-6;
const double PI = acos(-1);
typedef complex<double> Complex;
typedef vector<Complex> Poly;
Complex w[2][N];
void init() {
for (int i = 1; i < (1 << 18); i <<= 1) {
w[0][i] = w[1][i] = Complex(1, 0);
Complex wn(cos(PI / i), sin(PI / i));
for (int j = 1; j < i; j++)
w[1][i + j] = w[1][i + j - 1] * wn;
wn = Complex(cos(PI / i), -sin(PI / i));
for (int j = 1; j < i; j++)
w[0][i + j] = w[0][i + j - 1] * wn;
}
}
void transform(Complex *t, int len, int typ) {
for (int i = 0, j = 0, k; j < len; j++) {
if (i > j) swap(t[i], t[j]);
for (k = (len >> 1); k & i; k >>= 1) i ^= k;
i ^= k;
}
for (int i = 1; i < len; i <<= 1) {
for (int j = 0; j < len; j += (i << 1)) {
for (int k = 0; k < i; k++) {
Complex x = t[j + k], y = w[typ][i + k] * t[j + k + i];
t[j + k] = x + y;
t[j + k + i] = x - y;
}
}
}
if (typ) return;
for (int i = 0; i < len; i++)
t[i] = Complex(t[i].real() / (double) len, t[i].imag());
}
bool equ0(double x) {
return fabs(x) < eps;
}
Poly mul(const Poly a, const Poly b) {
static Poly cura, curb;
cura = a, curb = b;
int len = 1 << (int) ceil(log2(cura.size() + curb.size() - 1));
cura.resize(len), curb.resize(len);
transform(&cura[0], len, 1);
transform(&curb[0], len, 1);
for (int i = 0; i < len; i++)
cura[i] *= curb[i];
transform(&cura[0], len, 0);
return cura;
}
Poly ansadd(N, 0), anssub(N, 0);
int n, m, q, maxa, maxb;
int a[N], b[N];
int cnta[N], cntb[N];
void devide(int l, int r) {
if (l == r) return;
int mid = (l + r) >> 1;
devide(l, mid);
devide(mid + 1, r);
Poly x, y;
int sizl = mid - l + 1, sizr = r - mid;
x.resize(sizl);
y.resize(sizr);
for (int i = 0; i < sizl; i++)
x[i] = cnta[i + l];
for (int i = 0; i < sizr; i++)
y[i] = cntb[i + mid + 1];
x = mul(x, y);
for (int i = 0; i < (signed) x.size(); i++)
ansadd[i + l + mid + 1] += x[i];
}
void calcadd() {
devide(0, max(maxa, maxb));
}
void calcsub() {
static Poly x, y;
x.resize(maxb + maxa + 1);
y.resize(maxb + maxa + 1);
for (int i = 0; i <= maxb + maxa; i++)
x[i] = y[i] = 0;
for (int i = 0; i <= maxa; i++)
x[maxb + i] = cnta[i];
for (int i = 0; i <= maxb; i++)
y[maxb - i] = cntb[i];
x = mul(x, y);
for (int i = maxb * 2; i < (signed) x.size(); i++)
anssub[i - maxb * 2] = x[i];
}
void solve() {
n = read_int(), m = read_int(), q = read_int();
maxa = maxb = 0;
for (int i = 1; i <= n; i++) {
cnta[a[i] = read_int()]++;
maxa = max(maxa, a[i]);
}
for (int i = 1; i <= m; i++) {
cntb[b[i] = read_int()]++;
maxb = max(maxb, b[i]);
}
calcsub();
calcadd();
while (q--) {
int c = read_int();
print_ll((ll) round(ansadd[c].real()) + (ll) round(anssub[c].real()));
print_char('\n');
}
for (int i = 0; i <= maxa; i++)
cnta[i] = 0;
for (int i = 0; i <= maxb; i++)
cntb[i] = 0;
for (int i = 0; i <= maxa + maxb; i++)
ansadd[i] = anssub[i] = 0;
}
int main() {
#ifdef dream_maker
freopen("input.txt", "r", stdin);
#endif
init();
int T = read_int();
while (T--) solve();
return 0;
}
BZOJ4836: [Lydsy1704月赛]二元运算【分治FFT】【卡常(没卡过)】的更多相关文章
- bzoj 4836 [Lydsy1704月赛]二元运算 分治FFT+生成函数
[Lydsy1704月赛]二元运算 Time Limit: 8 Sec Memory Limit: 128 MBSubmit: 577 Solved: 201[Submit][Status][Di ...
- BZOJ4836 [Lydsy1704月赛]二元运算 分治 多项式 FFT
原文链接http://www.cnblogs.com/zhouzhendong/p/8830036.html 题目传送门 - BZOJ4836 题意 定义二元运算$opt$满足 $$x\ opt\ y ...
- BZOJ 4836: [Lydsy1704月赛]二元运算 分治FFT
Code: #include<bits/stdc++.h> #define ll long long #define maxn 500000 #define setIO(s) freope ...
- BZOJ4836: [Lydsy1704月赛]二元运算
BZOJ4836: [Lydsy1704月赛]二元运算 https://lydsy.com/JudgeOnline/problem.php?id=4836 分析: 分开做,维护两个桶. 分治每次求\( ...
- 【bzoj4836】[Lydsy2017年4月月赛]二元运算 分治+FFT
题目描述 定义二元运算 opt 满足 现在给定一个长为 n 的数列 a 和一个长为 m 的数列 b ,接下来有 q 次询问.每次询问给定一个数字 c 你需要求出有多少对 (i, j) 使得 a_ ...
- bzoj 4836: [Lydsy2017年4月月赛]二元运算 -- 分治+FFT
4836: [Lydsy2017年4月月赛]二元运算 Time Limit: 8 Sec Memory Limit: 128 MB Description 定义二元运算 opt 满足 现在给定一 ...
- [BZOJ4836]二元运算(分治FFT)
4836: [Lydsy1704月赛]二元运算 Time Limit: 8 Sec Memory Limit: 128 MBSubmit: 578 Solved: 202[Submit][Stat ...
- 【bzoj4836】二元运算 分治FFT
Description 定义二元运算 opt 满足 现在给定一个长为 n 的数列 a 和一个长为 m 的数列 b ,接下来有 q 次询问.每次询问给定一个数字 c 你需要求出有多少对 (i, j) 使 ...
- LOJ575. 「LibreOJ NOI Round #2」不等关系 [容斥,分治FFT]
LOJ 思路 发现既有大于又有小于比较难办,使用容斥,把大于改成任意减去小于的. 于是最后的串就长成这样:<<?<?<??<<<?<.我们把一段连续的& ...
随机推荐
- 一个很棒的Flutter学习资源列表
目录 文章 一开始 HOWTO文档 网站/博客 高级 视频 组件 演示 UI 材料设计 图片 地图 图表 导航 验证 文字和富文本 分析.流量统计 自动构建 风格样式 媒体 音频 视频 语音 存储 获 ...
- react notes
jsx 在JSX中嵌入用户输入是安全的,默认情况下, 在渲染之前, React DOM 会格式化(escapes) JSX中的所有值. 从而保证用户无法注入任何应用之外的代码. 在被渲染之前,所有的数 ...
- Java JDK5新特性-泛型
2017-10-30 22:47:11 Java 泛型(generics)是 JDK 5 中引入的一个新特性, 泛型提供了编译时类型安全检测机制,该机制允许程序员在编译时检测到非法的类型. 泛型的本质 ...
- python模块——random模块(简单验证码实现)
实现一个简单的验证码生成器 #!/usr/bin/env python # -*- coding:utf-8 -*- __author__ = "loki" # Usage: 验证 ...
- LeetCode--112--路径总和
问题描述: 给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和. 说明: 叶子节点是指没有子节点的节点. 示例: 给定如下二叉树,以及目标和 s ...
- android沉浸状态栏和顶部状态栏背景色的设置
法一: 现在很多应用都引用了沉浸式状态栏,如QQ,效果下图: 这样的效果很酷炫,其实设置很简单. 不过要说明的是,这种效果只能在API19以及以上版本中才能够做到. 如果想让界面Activity中实现 ...
- Memory and Casinos CodeForces - 712E (概率,线段树)
题目链接 题目大意:$n$个点, 每个点$i$有成功率$p_i$, 若成功走到$i+1$, 否则走到走到$i-1$, 多组询问, 求从$l$出发, 在$l$处不失败, 最后在$r$处胜利的概率 设$L ...
- RAC配置(启停库)
关库顺序 :先关闭数据库 然后关闭节点资源 [root@rac1 ~]# srvctl stop database -d 数据库名[root@rac1 ~]# srvctl stop ins ...
- Leetcode 96
class Solution { public: int numTrees(int n) { ]; dp[] = ; dp[] = ; dp[] = ; ;i <= n;i++){ ; ;j & ...
- 43. Multiply Strings 字符串表示的大数乘法
Given two numbers represented as strings, return multiplication of the numbers as a string. Note: Th ...