結果

問題 No.206 数の積集合を求めるクエリ
ユーザー sugarrrsugarrr
提出日時 2020-04-02 15:37:31
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 260 ms / 7,000 ms
コード長 3,249 bytes
コンパイル時間 1,695 ms
コンパイル使用メモリ 133,680 KB
実行使用メモリ 39,336 KB
最終ジャッジ日時 2023-08-20 14:33:54
合計ジャッジ時間 7,192 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 8 ms
4,512 KB
testcase_07 AC 7 ms
4,540 KB
testcase_08 AC 7 ms
4,524 KB
testcase_09 AC 8 ms
4,612 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 2 ms
4,380 KB
testcase_12 AC 8 ms
4,500 KB
testcase_13 AC 7 ms
4,508 KB
testcase_14 AC 8 ms
4,528 KB
testcase_15 AC 8 ms
4,460 KB
testcase_16 AC 8 ms
4,488 KB
testcase_17 AC 252 ms
39,192 KB
testcase_18 AC 249 ms
39,196 KB
testcase_19 AC 252 ms
39,208 KB
testcase_20 AC 239 ms
39,240 KB
testcase_21 AC 248 ms
39,196 KB
testcase_22 AC 245 ms
39,184 KB
testcase_23 AC 260 ms
39,328 KB
testcase_24 AC 256 ms
39,260 KB
testcase_25 AC 252 ms
39,248 KB
testcase_26 AC 252 ms
39,224 KB
testcase_27 AC 257 ms
39,200 KB
testcase_28 AC 255 ms
39,328 KB
testcase_29 AC 253 ms
39,332 KB
testcase_30 AC 252 ms
39,336 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
//#include "boost/multiprecision/cpp_int.hpp"
//typedef boost::multiprecision::cpp_int LL;
typedef long double dd;
//#define i_7 (ll)(1E9+7)
#define i_7 998244353
#define i_5 i_7-2

ll mod(ll a){
    ll c=a%i_7;
    if(c>=0)return c;
    return c+i_7;
}
typedef pair<ll,ll> l_l;
typedef pair<dd,dd> d_d;
ll inf=(ll)1E16;
#define rep(i,l,r) for(ll i=l;i<=r;i++)
#define pb push_back
ll max(ll a,ll b){if(a<b)return b;else return a;}
ll min(ll a,ll b){if(a>b)return b;else return a;}
void Max(ll &pos,ll val){pos=max(pos,val);}//Max(dp[n],dp[n-1]);
void Min(ll &pos,ll val){pos=min(pos,val);}
void Add(ll &pos,ll val){pos=mod(pos+val);}
dd EPS=1E-9;
#define fastio ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define fi first
#define se second
#define endl "\n"
#define SORT(v) sort(v.begin(),v.end())
#define ERASE(v) v.erase(unique(v.begin(),v.end()),v.end())
#define POSL(v,x) (lower_bound(v.begin(),v.end(),x)-v.begin())
#define POSU(v,x) (upper_bound(v.begin(),v.end(),x)-v.begin())
//template<class T>void max(T a,T b){if(a<b)return b;else return a;}
//template<class T>void min(T a,T b){if(a>b)return b;else return a;}
//template<class T>bool Max(T&a, T b){if(a < b){a = b;return 1;}return 0;}
//template<class T>bool Min(T&a, T b){if(a > b){a = b;return 1;}return 0;}

//////////////////////////

using Complex = complex<double>;
vector<Complex> dft(vector<Complex> A, int N, int sgn = 1) {
    if(N == 1) return A;
 
    vector<Complex> F(N / 2), G(N / 2);
    for(int i=0; i<N/2; i++) {
        F[i] = A[2 * i + 0];
        G[i] = A[2 * i + 1];
    }
 
    F = dft(F, N / 2, sgn);
    G = dft(G, N / 2, sgn);
 
    Complex zeta(cos(2.0 * M_PI / N), sin(2.0 * M_PI / N) * sgn);
    Complex pow_zeta = 1;
 
    for(int i=0; i<N; i++) {
        A[i] = F[i % (N / 2)] + pow_zeta * G[i % (N / 2)];
        pow_zeta *= zeta;
    }
    return A;
}
 
vector<Complex> inv_dft(vector<Complex> A, int N) {
    A = dft(A, N, -1);
    for(int i=0; i<N; i++) {
        A[i] /= N;
    }
    return A;
}
 
vector<Complex> multiply(vector<Complex> A, vector<Complex> B) {
    int sz = A.size() + B.size() + 1;
    int N = 1; while(N < sz) N *= 2;
 
    A.resize(N), B.resize(N);
    A = dft(A, N);
    B = dft(B, N);
 
    vector<Complex> F(N);
    for(int i=0; i<N; i++) {
        F[i] = A[i] * B[i];
    }
    return inv_dft(F, N);
}


////////////////////////////
/*
int main() {
    int N; scanf("%d", &N);
 
    vector<Complex> A(N+1), B(N+1);
    A[0] = B[0] = Complex(0);
    for(int i=1; i<=N; i++) {
        int vl, vr; scanf("%d%d", &vl, &vr);
        A[i] = Complex(vl);
        B[i] = Complex(vr);
    }
 
    vector<Complex> ans = multiply(A, B);
    for(int i=1; i<=2*N; i++) {
        printf("%d\n", (int)(ans[i].real() + 0.5));///////////////こうやって使う
    }
    return 0;
}
*/

int main(){fastio
    ll l,m,n;cin>>l>>m>>n;
    vector<Complex>a(n+1),b(n+1);
    rep(x,1,l){
        ll c;cin>>c;
        a[c]=1;
    }
    rep(x,1,m){
        ll c;cin>>c;
        b[n-c]=1;
    }
    ll q;cin>>q;
    vector<Complex>ans=multiply(a,b);
    rep(i,n,n+q-1){
        cout<<(ll)(ans[i].real()+0.5)<<endl;
    }
    
    return 0;
}
0