結果

問題 No.206 数の積集合を求めるクエリ
ユーザー chikkun0310chikkun0310
提出日時 2020-04-02 15:40:32
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 310 ms / 7,000 ms
コード長 2,405 bytes
コンパイル時間 1,939 ms
コンパイル使用メモリ 177,688 KB
実行使用メモリ 40,792 KB
最終ジャッジ日時 2023-09-11 00:26:46
合計ジャッジ時間 8,260 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 9 ms
4,476 KB
testcase_07 AC 9 ms
4,528 KB
testcase_08 AC 9 ms
4,468 KB
testcase_09 AC 9 ms
4,472 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 2 ms
4,376 KB
testcase_12 AC 10 ms
4,708 KB
testcase_13 AC 10 ms
4,472 KB
testcase_14 AC 9 ms
4,480 KB
testcase_15 AC 9 ms
4,584 KB
testcase_16 AC 9 ms
4,648 KB
testcase_17 AC 308 ms
40,792 KB
testcase_18 AC 288 ms
40,016 KB
testcase_19 AC 302 ms
40,576 KB
testcase_20 AC 288 ms
39,952 KB
testcase_21 AC 292 ms
40,288 KB
testcase_22 AC 292 ms
40,152 KB
testcase_23 AC 301 ms
40,628 KB
testcase_24 AC 310 ms
40,732 KB
testcase_25 AC 308 ms
40,612 KB
testcase_26 AC 294 ms
39,952 KB
testcase_27 AC 288 ms
39,808 KB
testcase_28 AC 300 ms
40,284 KB
testcase_29 AC 298 ms
40,220 KB
testcase_30 AC 295 ms
39,868 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//#include <boost/multiprecision/cpp_ll.hpp>
//typedef boost::multiprecision::cpp_ll 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;
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;
string alp="abcdefghijklmnopqrstuvwxyz";
#define endl "\n"
#define fastio ios::sync_with_stdio(false); cin.tie(0); cout.tie(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(){fastio
    ll l,m,n;cin>>l>>m>>n;
    ll a[l],b[m];
    ll q;
    rep(i,0,l-1)cin>>a[i];
    rep(i,0,m-1)cin>>b[i];
    cin>>q;
    sort(a,a+l);
    sort(b,b+m,greater<ll>());
    ll c=b[0];
    rep(i,0,m-1){
        b[i]=-b[i]+c;
    }
    vector<Complex> A(n+1),B(n+1);
    rep(i,0,n){
        A[i]=0;
        B[i]=0;
    }
    rep(i,0,l-1){
        A[a[i]]=1;
    }
    rep(i,0,m-1){
        B[b[i]]=1;
    }
    vector<Complex> C;
    C=multiply(A,B);
    rep(v,0,q-1){
        cout<<(ll)(C[v+c].real()+0.5)<<endl;
    }
    
    
    return 0;
}
0