#287122 (C++11) No.206 数の積集合を求めるクエリ

提出ソース

結果

問題	No.206 数の積集合を求めるクエリ
ユーザー	mdj982
提出日時	2018-09-25 20:44:59
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	234 ms / 7,000 ms
コード長	4,185 bytes
コンパイル時間	963 ms
コンパイル使用メモリ	112,572 KB
実行使用メモリ	34,124 KB
最終ジャッジ日時	2024-10-05 18:20:56
合計ジャッジ時間	5,174 ms
ジャッジサーバーID （参考情報）	judge3 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 6,816 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	2 ms 6,816 KB
testcase_03	AC	1 ms 6,816 KB
testcase_04	AC	1 ms 6,820 KB
testcase_05	AC	2 ms 6,816 KB
testcase_06	AC	4 ms 6,816 KB
testcase_07	AC	4 ms 6,820 KB
testcase_08	AC	4 ms 6,816 KB
testcase_09	AC	4 ms 6,816 KB
testcase_10	AC	1 ms 6,816 KB
testcase_11	AC	2 ms 6,816 KB
testcase_12	AC	7 ms 6,820 KB
testcase_13	AC	7 ms 6,820 KB
testcase_14	AC	8 ms 6,820 KB
testcase_15	AC	7 ms 6,816 KB
testcase_16	AC	7 ms 6,820 KB
testcase_17	AC	122 ms 33,940 KB
testcase_18	AC	114 ms 33,920 KB
testcase_19	AC	121 ms 34,124 KB
testcase_20	AC	118 ms 34,016 KB
testcase_21	AC	118 ms 34,012 KB
testcase_22	AC	116 ms 34,116 KB
testcase_23	AC	125 ms 34,072 KB
testcase_24	AC	234 ms 33,936 KB
testcase_25	AC	227 ms 34,068 KB
testcase_26	AC	225 ms 34,060 KB
testcase_27	AC	191 ms 34,072 KB
testcase_28	AC	223 ms 33,936 KB
testcase_29	AC	221 ms 33,924 KB
testcase_30	AC	219 ms 34,052 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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); }

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

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;
}

// a.size() should be 2^digit
vector<cdouble> FFT(const vector<cdouble>& a) {
	int n = int(a.size());
	int digit = int(rndf(log2(n)));
	vector<cdouble> ret(n);
	vector<cdouble> w(n >> 1);
	Loop(i, n >> 1) w[i] = exp(cdouble({ 0, 2 * PI * i / n }));
	Loop(i, n) ret[i] = a[bit_reverse(i, digit)];
	Loop(i, digit) {
		int j = 0, m = 1 << i, mw = (digit - i - 1);
		Loop(group_id, n >> (i + 1)) {
			Loop(k, m) {
				cdouble x = ret[j] + w[k << mw] * ret[j + m];
				cdouble y = ret[j] - w[k << mw] * ret[j + m];
				ret[j] = x; ret[j + m] = y;
				++j;
			}
			j += m;
		}
	}
	return ret;
}


// f.size() should be 2^digit
vector<cdouble> IFFT(const vector<cdouble>& f) {
	int n = int(f.size());
	int digit = int(rndf(log2(n)));
	vector<cdouble> a = f;
	vector<cdouble> w(n >> 1);
	Loop(i, n >> 1) w[i] = exp(cdouble({ 0, 2 * PI * i / 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 = (a[j] + a[j + m]) / 2.0;
				cdouble r = (a[j] - a[j + m]) / (w[k << mw] * 2.0);
				a[j] = q; a[j + m] = r;
				++j;
			}
			j += m;
		}
	}
	vector<cdouble> ret(n);
	Loop(i, n) ret[i] = a[bit_reverse(i, digit)];
	return ret;
}

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

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;
	}
	vector<cdouble> g = FFT(a);
	vector<cdouble> h = FFT(b);
	vector<cdouble> gh(n);
	Loop(i, gh.size()) {
		gh[i] = g[i] * h[i];
	}
	vector<cdouble> c = IFFT(gh);
	int q; cin >> q;
	Loop(i, q) {
		cout << rndf(c[i].real()) << endl;
	}
}

yukicoder

結果

テストケース

ソースコード