結果

問題 No.206 数の積集合を求めるクエリ
コンテスト
ユーザー Ryuhei Mori
提出日時 2017-03-02 10:39:24
言語 C90
(gcc 12.3.0)
結果
AC  
実行時間 30 ms / 7,000 ms
コード長 2,381 bytes
コンパイル時間 582 ms
コンパイル使用メモリ 28,656 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2024-06-26 03:14:43
合計ジャッジ時間 3,124 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <complex.h>

#define PI2 6.28318530717958647692

const int k = 18;
const int m = (1 << 18);

typedef double complex cmplx;

cmplx C[1<<18];
cmplx w[1<<17];

void read_int(int *x){
  char c=0;
  while(c<'0'||c>'9') c = getchar();

  *x = c-'0';
  c = getchar();

  while(c>='0'&&c<='9') {
    *x *= 10;
    *x += c-'0';
    c = getchar();
  }
}

void write_int(int x){
  int i;
  static char buf[12];
  if(x==0){
    puts("0");
    return;
  }

  for(i=10; x; i--){
    buf[i] = '0' + x % 10;
    x /= 10;
  }
  puts(buf+1+i);
}

unsigned minus(unsigned x) {
  int y = 1 << (31-__builtin_clz(x));
  return (~x&(y-1))^y;
}


void fft(cmplx *A, int k){
  int m = 1 << k;
  int u = 1;
  int v = m/2;
  int i, jh, j, je;
  for(i=k;i>0;i--){
    for(jh=0;jh<u;jh++){
      cmplx wj = w[jh];
      for(j = jh << i, je = j|v;j<je; j++){
        cmplx Ajv = wj * A[j|v];
        A[j|v] = A[j] - Ajv;
        A[j] += Ajv;
      }
    }
    u <<= 1;
    v >>= 1;
  }
}

void ifft(cmplx *A, int k){
  int m = 1 << k;
  int u = m/2;
  int v = 1;
  int i, jh, j, je;
  for(i=1;i<=k;i++){
    for(jh=0;jh<u;jh++){
      cmplx wj = conj(w[jh]);
      for(j = jh << i, je = j|v;j<je; j++){
        cmplx Ajv = A[j] - A[j|v];
        A[j] += A[j|v];
        A[j|v] = wj * Ajv;
      }
    }
    u >>= 1;
    v <<= 1;
  }
}

void genw(int i, int b, cmplx z){
  if(b == 0){
    w[i] = z;
  }
  else {
    genw(i, b>>1, z);
    genw(i|b, b>>1, z*w[b]);
  }
}

int main(){
  int l, mm, n, q;
  int i, j;
  const double arg = -PI2/m;

  for(i=1, j=m/4; j; i<<=1, j>>=1){
    w[i] = cexp(I * (arg * j));
  }
  genw(0, m/4, 1);

  read_int(&l);
  read_int(&mm);
  read_int(&n);


  for(i=0;i<l;i++){
    int a;
    read_int(&a);
    C[a] = 1;
  }
  for(i=0;i<mm;i++){
    int a;
    read_int(&a);
    C[(1<<18)-a] += I;
  }
  read_int(&q);

  fft(C, k);

  C[0] = 4 * creal(C[0]) * cimag(C[0]) * I;
  C[1] = 4 * creal(C[1]) * cimag(C[1]) * I;

  for(i = 2; i < m; i+=2){
    int j = minus(i);
    C[i] = (C[i] + conj(C[j]))*(C[i] - conj(C[j]));
    C[j] = -conj(C[i]);
  }

  for(i = 0; i < m; i+=2){
    C[i/2] = C[i]+C[i^1] + (C[i]-C[i^1])*w[i/2]*I;
  }

  ifft(C, k-1);

  write_int(cimag(C[0])/(4*m)+0.5);
  for(i=1;2*i<q;i++){
    write_int(-creal(C[i])/(4*m)+0.5);
    write_int(cimag(C[i])/(4*m)+0.5);
  }
  if(!(q&1)) write_int(-creal(C[q/2])/(4*m)+0.5);

  return 0;
}
0