結果

問題 No.1036 Make One With GCD 2
ユーザー Haar
提出日時 2020-05-20 18:30:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,473 bytes
コンパイル時間 2,588 ms
コンパイル使用メモリ 202,676 KB
最終ジャッジ日時 2025-01-10 13:25:38
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>


template <typename Semilattice>
class SparseTable{
  using value_type = typename Semilattice::value_type;
  
  std::vector<std::vector<value_type>> a;
  std::vector<int> log_table;
  
public:
  SparseTable(const std::vector<value_type> &v){
    int n = v.size();
    int logn = 0;
    while((1 << logn) <= n) ++logn;
    
    a.assign(n, std::vector<value_type>(logn));
    for(int i = 0; i < n; ++i) a[i][0] = v[i];
    for(int j = 1; j < logn; ++j){
      for(int i = 0; i < n; ++i){
        a[i][j] = Semilattice::op(a[i][j-1], a[std::min<int>(n-1, i+(1<<(j-1)))][j-1]);
      }
    }

    log_table.assign(n+1, 0);
    for(int i = 2; i < n+1; ++i) log_table[i] = log_table[i>>1] + 1;
  }
  
  inline value_type get(int s, int t) const { // [s,t)
    int k = log_table[t-s];
    return Semilattice::op(a[s][k], a[t-(1<<k)][k]);
  }

  inline value_type get(std::vector<std::pair<int,int>> st) const {
    value_type ret;
    bool t = true;

    for(const auto &p : st){
      if(p.first < p.second){
        if(t){
          ret = get(p.first, p.second);
          t = false;
        }else{
          ret = Semilattice::op(ret, get(p.first, p.second));
        }
      }
    }

    return ret;
  }
};


int64_t binary_gcd(int64_t a, int64_t b){
  int64_t g = 1;

  while(1){
    if(a > b) std::swap(a, b);
    
    if(a == 0){
      break;
    }else{
      if((a & 1) == 0 and (b & 1) == 0){
        a >>= 1;
        b >>= 1;
        g <<= 1;
      }else if((a & 1) == 0){
        a >>= 1;
      }else if((b & 1) == 0){
        b >>= 1;
      }else{
        int64_t t = std::abs(a - b) >> 1;
        b = t;
      }
    }
  }

  return g * b;
}






template <typename T>
struct GcdMonoid{
  using value_type = T;
  constexpr inline static value_type id(){return 0;}
  constexpr inline static value_type op(const value_type &a, const value_type &b){return binary_gcd(a, b);}
};

int main(){
  std::cin.tie(0);
  std::ios::sync_with_stdio(false);
  
  int N;

  while(std::cin >> N){
    std::vector<int64_t> A(N);
    for(int i = 0; i < N; ++i) std::cin >> A[i];

    SparseTable<GcdMonoid<int64_t>> s(A);

    int64_t ans = 0;

    for(int l = 0; l < N; ++l){
      int lb = l, ub = N+1;
      while(abs(lb-ub) > 1){
        int mid = (lb + ub) / 2;

        if(s.get(l, mid) == 1){
          ub = mid;
        }else{
          lb = mid;
        }
      }
      
      ans += (N - lb);
    }

    std::cout << ans << "\n";
  }

  return 0;
}
0