結果

問題 No.206 数の積集合を求めるクエリ
ユーザー mdj982mdj982
提出日時 2018-09-26 02:51:59
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 224 ms / 7,000 ms
コード長 4,316 bytes
コンパイル時間 1,105 ms
コンパイル使用メモリ 110,820 KB
実行使用メモリ 15,716 KB
最終ジャッジ日時 2024-04-16 20:27:21
合計ジャッジ時間 4,839 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 7 ms
5,376 KB
testcase_13 AC 7 ms
5,376 KB
testcase_14 AC 8 ms
5,376 KB
testcase_15 AC 7 ms
5,376 KB
testcase_16 AC 8 ms
5,376 KB
testcase_17 AC 101 ms
15,616 KB
testcase_18 AC 92 ms
15,680 KB
testcase_19 AC 100 ms
15,608 KB
testcase_20 AC 93 ms
15,616 KB
testcase_21 AC 95 ms
15,616 KB
testcase_22 AC 96 ms
15,688 KB
testcase_23 AC 99 ms
15,616 KB
testcase_24 AC 224 ms
15,612 KB
testcase_25 AC 216 ms
15,612 KB
testcase_26 AC 221 ms
15,616 KB
testcase_27 AC 169 ms
15,616 KB
testcase_28 AC 214 ms
15,716 KB
testcase_29 AC 219 ms
15,704 KB
testcase_30 AC 221 ms
15,632 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <fstream>
#include <iomanip>
#include <climits>
#include <limits>
#include <algorithm>
#include <array>
#include <vector>
#include <deque>
#include <queue>
#include <list>
#include <stack>
#include <string>
#include <functional>
#include <numeric>
#include <map>
#include <set>
#include <cstdlib>
#include <bitset>
#include <unordered_map>
#include <random>
#include <cmath>
#include <complex>
// #include "utiltime.hpp"

using namespace std;

typedef long long int ll;
typedef vector<int> vi;
typedef vector<vector<int>> vvi;
typedef pair<int, int> P;
typedef pair<ll, ll> Pll;
typedef vector<ll> vll;
typedef vector<vector<ll>> vvll;
typedef complex<double> cdouble;

const int INFL = (int)1e9;
const ll INFLL = (ll)1e18;
const double INFD = numeric_limits<double>::infinity();
const double PI = 3.14159265358979323846;
#define Loop(i, n) for(int i = 0; i < (int)n; i++)
#define Loopll(i, n) for(ll i = 0; i < (ll)n; i++)
#define Loop1(i, n) for(int i = 1; i <= (int)n; i++)
#define Loopll1(i, n) for(ll i = 1; i <= (ll)n; i++)
#define Loopr(i, n) for(int i = (int)n - 1; i >= 0; i--)
#define Looprll(i, n) for(ll i = (ll)n - 1; i >= 0; i--)
#define Loopr1(i, n) for(int i = (int)n; i >= 1; i--)
#define Looprll1(i, n) for(ll i = (ll)n; i >= 1; i--)
#define Loopitr(itr, container) for(auto itr = container.begin(); itr != container.end(); itr++)
#define printv(vector) Loop(i, vector.size()) { cout << vector[i] << " "; } cout << endl;
#define printmx(matrix) Loop(i, matrix.size()) { Loop(j, matrix[i].size()) { cout << matrix[i][j] << " "; } cout << endl; }
#define quickio() ios::sync_with_stdio(false); cin.tie(0);
#define bitmanip(m,val) static_cast<bitset<(int)m>>(val)
ll rndf(double x) { return (ll)(x + (x >= 0 ? 0.5 : -0.5)); }
ll floorsqrt(double x) { ll m = (ll)sqrt(x); return m + (m * m <= (ll)(x) ? 0 : -1); }
ll ceilsqrt(double x) { ll m = (ll)sqrt(x); return m + ((ll)x <= m * m ? 0 : 1); }
ll rnddiv(ll a, ll b) { return (a / b + (a % b * 2 >= b ? 1 : 0)); }
ll ceildiv(ll a, ll b) { return (a / b + (a % b == 0 ? 0 : 1)); }
ll gcd(ll m, ll n) { if (n == 0) return m; else return gcd(n, m % n); }

/*******************************************************/

namespace Fourier_transform {

	vector<cdouble> ws, iws;

	inline int bit_reverse(int x, int digit) {
		int ret = digit ? x & 1 : 0;
		Loop(i, digit - 1) { ret <<= 1; x >>= 1; ret |= x & 1; }
		return ret;
	}

	inline void make_ws(int n) {
		if (ws.size() != (n >> 1)) {
			ws.resize(n >> 1);
			Loop(i, n >> 1) ws[i] = exp(cdouble({ 0, 2 * PI * i / n }));
		}
	}

	inline void make_iws(int n) {
		if (iws.size() != (n >> 1)) {
			iws.resize(n >> 1);
			Loop(i, n >> 1) iws[i] = exp(cdouble({ 0, -2 * PI * i / n }));
		}
	}

	// a.size() should be 2^digit
	void FFT(vector<cdouble>& a) {
		int n = int(a.size());
		int digit = int(rndf(log2(n)));
		make_ws(n);
		Loop(i, n) {
			int j = bit_reverse(i, digit);
			if (j > i) swap(a[i], a[j]);
		}
		Loop(i, digit) {
			int j = 0, m = 1 << i, mw = (digit - i - 1);
			Loop(group_id, n >> (i + 1)) {
				Loop(k, m) {
					cdouble x = a[j] + ws[k << mw] * a[j + m];
					cdouble y = a[j] - ws[k << mw] * a[j + m];
					a[j] = x; a[j + m] = y;
					++j;
				}
				j += m;
			}
		}
	}

	// f.size() should be 2^digit
	void IFFT(vector<cdouble>& f) {
		int n = int(f.size());
		int digit = int(rndf(log2(n)));
		make_iws(n);
		Loopr(i, digit) {
			int j = 0, m = 1 << i, mw = (digit - i - 1);
			Loop(group_id, n >> (i + 1)) {
				Loop(k, m) {
					cdouble q = (f[j] + f[j + m]) * 0.5;
					cdouble r = (f[j] - f[j + m]) * 0.5 * iws[k << mw];
					f[j] = q; f[j + m] = r;
					++j;
				}
				j += m;
			}
		}
		Loop(i, n) {
			int j = bit_reverse(i, digit);
			if (j > i) swap(f[i], f[j]);
		}
	}

	int legal_size_of(int n) {
		int ret = 1 << (int)log2(n);
		if (ret < n) ret <<= 1;
		return ret;
	}

}

using namespace Fourier_transform;

int main() {
	quickio();
	int L, M, N; cin >> L >> M >> N;
	int n = legal_size_of(N * 2 - 1);
	vector<cdouble> a(n, 0), b(n, 0);
	Loop(i, L) {
		int abuf; cin >> abuf;
		abuf--;
		a[abuf] = 1;
	}
	Loop(i, M) {
		int bbuf; cin >> bbuf;
		bbuf--;
		if (bbuf == 0) b[0] = 1;
		else b[n - bbuf] = 1;
	}
	FFT(a);
	FFT(b);
	Loop(i, n) a[i] *= b[i];
	IFFT(a);
	int q; cin >> q;
	Loop(i, q) {
		cout << rndf(a[i].real()) << endl;
	}
}
0