結果

問題 No.206 数の積集合を求めるクエリ
コンテスト
ユーザー EmKjp
提出日時 2015-05-08 22:57:30
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 130 ms / 7,000 ms
コード長 2,254 bytes
コンパイル時間 740 ms
コンパイル使用メモリ 93,656 KB
実行使用メモリ 17,392 KB
最終ジャッジ日時 2024-07-05 19:14:06
合計ジャッジ時間 3,735 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:82:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   82 |     FOR(i,L) scanf("%d",&A[i]);
      |              ~~~~~^~~~~~~~~~~~
main.cpp:83:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   83 |     FOR(i,M) scanf("%d",&B[i]);
      |              ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include<iostream>
#include<sstream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<numeric>
#include<functional>
#include<complex>

using namespace std;
#define BET(a,b,c) ((a)<=(b)&&(b)<(c))
#define FOR(i,n) for(int i=0,i##_end=(int(n));i<i##_end;i++)
#define SZ(x) (int)(x.size())
#define ALL(x) (x).begin(),(x).end()
#define MP make_pair
#define FOR_EACH(it,v) for(__typeof(v.begin()) it=v.begin(),it_end=v.end() ; it != it_end ; it++)
typedef vector<int> VI;
typedef vector<VI> VVI;


const long double PI = 4.0*atan(1.0);

typedef complex<double> Complex;
const Complex I(0, 1);

template<typename T>
void fft(int n, double theta, vector<T>& a) {
  for (int m = n; m >= 2; m >>= 1) {
    int mh = m >> 1;
    for (int i = 0; i < mh; i++) {
      Complex w = exp(i*theta*I);
      for (int j = i; j < n; j += m) {
        int k = j + mh;
        Complex x = a[j] - a[k];
        a[j] += a[k];
        a[k] = w * x;
      }
    }
    theta *= 2;
  }
  int i = 0;
  for (int j = 1; j < n - 1; j++) {
    for (int k = n >> 1; k > (i ^= k); k >>= 1);
    if (j < i) swap(a[i], a[j]);
  }
}

// result[i] = sum a[x] * b[y] where x + y == i
template<typename T>
vector<long long> comvolution(const vector<T> &a, const vector<T> &b){
    vector<long long> result;
    vector<Complex> fa(a.begin(), a.end());
    vector<Complex> fb(b.begin(), b.end());
    //while(SZ(fa) && fa.back().real() == 0.0) fa.pop_back();
    //while(SZ(fb) && fb.back().real() == 0.0) fb.pop_back();
    int n = 1;
    while(n < SZ(fa) + SZ(fb)) n <<= 1;
    fa.resize(n);
    fb.resize(n);
    fft(n, 2 * PI / n, fa);
    fft(n, 2 * PI / n, fb);
    FOR(i,n) fa[i] *= fb[i];
    fft(n, -2 * PI / n, fa);
    FOR(i,n) result.push_back(round(fa[i].real()/n));
    return result;
}


int main()
{
    int L,M,N;
    cin>>L>>M>>N;
    VI A(L);
    VI B(M);
    FOR(i,L) scanf("%d",&A[i]);
    FOR(i,M) scanf("%d",&B[i]);
    int Q;
    cin>>Q;
    VI AH(N + 1);
    VI BH(N + 1);
    FOR(i,L) AH[A[i]]++;
    FOR(i,M) BH[B[i]]++;
    reverse(ALL(BH));
    auto r = comvolution(AH, BH);
    FOR(i,Q){
        printf("%lld\n", r[i+N]);
    }
    return 0;
}
0