結果

問題 No.983 Convolution
ユーザー LayCurseLayCurse
提出日時 2020-02-11 14:09:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,334 bytes
コンパイル時間 2,460 ms
コンパイル使用メモリ 215,164 KB
実行使用メモリ 12,824 KB
最終ジャッジ日時 2024-10-01 07:30:59
合計ジャッジ時間 4,163 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 20 ms
9,216 KB
testcase_04 AC 16 ms
8,064 KB
testcase_05 AC 15 ms
7,424 KB
testcase_06 AC 22 ms
10,112 KB
testcase_07 AC 22 ms
9,984 KB
testcase_08 AC 6 ms
5,376 KB
testcase_09 WA -
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 7 ms
5,376 KB
testcase_12 AC 25 ms
12,032 KB
testcase_13 AC 9 ms
5,760 KB
testcase_14 AC 10 ms
6,272 KB
testcase_15 AC 18 ms
12,136 KB
testcase_16 AC 14 ms
12,004 KB
testcase_17 AC 22 ms
12,516 KB
testcase_18 AC 6 ms
8,008 KB
testcase_19 AC 15 ms
10,932 KB
testcase_20 AC 26 ms
12,824 KB
testcase_21 AC 20 ms
11,596 KB
testcase_22 AC 6 ms
9,676 KB
testcase_23 AC 17 ms
11,428 KB
testcase_24 AC 5 ms
7,956 KB
testcase_25 AC 5 ms
7,892 KB
testcase_26 AC 12 ms
10,828 KB
testcase_27 AC 10 ms
10,504 KB
testcase_28 AC 18 ms
11,000 KB
testcase_29 AC 32 ms
12,424 KB
testcase_30 AC 17 ms
11,980 KB
testcase_31 AC 11 ms
10,284 KB
testcase_32 AC 20 ms
11,072 KB
testcase_33 AC 3 ms
6,816 KB
testcase_34 AC 3 ms
7,628 KB
testcase_35 AC 2 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize ("Ofast")
#include<bits/stdc++.h>
using namespace std;
inline void rd(int &x){
  int k;
  int m=0;
  x=0;
  for(;;){
    k = getchar_unlocked();
    if(k=='-'){
      m=1;
      break;
    }
    if('0'<=k&&k<='9'){
      x=k-'0';
      break;
    }
  }
  for(;;){
    k = getchar_unlocked();
    if(k<'0'||k>'9'){
      break;
    }
    x=x*10+k-'0';
  }
  if(m){
    x=-x;
  }
}
inline void rd(long long &x){
  int k;
  int m=0;
  x=0;
  for(;;){
    k = getchar_unlocked();
    if(k=='-'){
      m=1;
      break;
    }
    if('0'<=k&&k<='9'){
      x=k-'0';
      break;
    }
  }
  for(;;){
    k = getchar_unlocked();
    if(k<'0'||k>'9'){
      break;
    }
    x=x*10+k-'0';
  }
  if(m){
    x=-x;
  }
}
inline void wt_L(char a){
  putchar_unlocked(a);
}
inline void wt_L(int x){
  int s=0;
  int m=0;
  char f[10];
  if(x<0){
    m=1;
    x=-x;
  }
  while(x){
    f[s++]=x%10;
    x/=10;
  }
  if(!s){
    f[s++]=0;
  }
  if(m){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
inline void wt_L(long long x){
  int s=0;
  int m=0;
  char f[20];
  if(x<0){
    m=1;
    x=-x;
  }
  while(x){
    f[s++]=x%10;
    x/=10;
  }
  if(!s){
    f[s++]=0;
  }
  if(m){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
template<class T, class U> inline T GCD_L(T a, U b){
  T r;
  while(b){
    r=a;
    a=b;
    b=r%a;
  }
  return a;
}
template<class T, class S> void ZetaTransform_L(int N, T A[], S res[] = NULL){
  int i;
  int j;
  if(res==NULL){
    for(i=0;(1<<i)<N;i++){
      for(j=(0);j<(N);j++){
        if(j&1<<i){
          A[j] += A[j^(1<<i)];
        }
      }
    }
  }
  else{
    for(j=(0);j<(N);j++){
      res[j] = A[j];
    }
    for(i=0;(1<<i)<N;i++){
      for(j=(0);j<(N);j++){
        if(j&1<<i){
          res[j] += res[j^(1<<i)];
        }
      }
    }
  }
}
template<class T> void MoebiusTransform(int N, T A[], T res[] = NULL){
  int i;
  int j;
  if(res==NULL){
    for(i=0;(1<<i)<N;i++){
      for(j=(0);j<(N);j++){
        if(j&1<<i){
          A[j] -= A[j^(1<<i)];
        }
      }
    }
  }
  else{
    for(j=(0);j<(N);j++){
      res[j] = A[j];
    }
    for(i=0;(1<<i)<N;i++){
      for(j=(0);j<(N);j++){
        if(j&1<<i){
          res[j] -= res[j^(1<<i)];
        }
      }
    }
  }
}
inline void wt_L(__int128_t x){
  int s=0;
  int m=0;
  char f[40];
  if(x<0){
    m=1;
    x=-x;
  }
  while(x){
    f[s++]=x%10;
    x/=10;
  }
  if(!s){
    f[s++]=0;
  }
  if(m){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
int N;
long long A[200000];
__int128_t b[200000];
__int128_t b2[200000];
int c[200000];
int c2[200000];
int main(){
  int i;
  __int128_t g;
  rd(N);
  {
    int cTE1_r3A;
    for(cTE1_r3A=(0);cTE1_r3A<(N);cTE1_r3A++){
      rd(A[cTE1_r3A]);
    }
  }
  for(i=(0);i<(N);i++){
    b[i] = 0;
  }
  for(i=(0);i<(N);i++){
    if(A[i]==-1){
      c[i]++;
    }
    else{
      b[i] = A[i];
    }
  }
  ZetaTransform_L(N,b,b2);
  ZetaTransform_L(N,c,c2);
  for(i=(0);i<(N);i++){
    b2[i] *= b2[i];
  }
  for(i=(0);i<(N);i++){
    c2[i] *= c2[i];
  }
  MoebiusTransform(N,b2,b);
  MoebiusTransform(N,c2,c);
  for(i=(0);i<(N);i++){
    if(c[i]){
      b[i] = 0;
    }
  }
  {
    int YREPHmFM;
    __int128_t jZyWAPpY;
    if(N==0){
      jZyWAPpY = 0;
    }
    else{
      jZyWAPpY = b[0];
      for(YREPHmFM=(1);YREPHmFM<(N);YREPHmFM++){
        jZyWAPpY = GCD_L(jZyWAPpY, b[YREPHmFM]);
      }
    }
    g =jZyWAPpY;
  }
  if(g==0){
    g = -1;
  }
  wt_L(g);
  wt_L('\n');
  return 0;
}
// cLay varsion 20200119-1

// --- original code ---
// inline void wt_L(__int128_t x){
//   int s=0, m=0;
//   char f[40];
//   if(x<0) m=1, x=-x;
//   while(x) f[s++]=x%10, x/=10;
//   if(!s) f[s++]=0;
//   if(m) putchar_unlocked('-');
//   while(s--) putchar_unlocked(f[s]+'0');
// }
// 
// int N; ll A[2d5];
// 
// __int128_t b[2d5], b2[2d5];
// int c[2d5], c2[2d5];
// {
//   __int128_t g;
// 
//   rd(N,A(N));
//   rep(i,N) b[i] = 0;
//   rep(i,N) if[A[i]==-1, c[i]++, b[i] = A[i]];
// 
//   ZetaTransform(N,b,b2);
//   ZetaTransform(N,c,c2);
//   rep(i,N) b2[i] *= b2[i];
//   rep(i,N) c2[i] *= c2[i];
//   MoebiusTransform(N,b2,b);
//   MoebiusTransform(N,c2,c);
// 
//   rep(i,N) if(c[i]) b[i] = 0;
//   g = gcd(b(N));
//   if(g==0) g = -1;
//   wt(g);
// }
0