結果
問題 | No.1036 Make One With GCD 2 |
ユーザー |
![]() |
提出日時 | 2020-04-26 17:06:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,527 ms / 2,000 ms |
コード長 | 3,323 bytes |
コンパイル時間 | 4,575 ms |
コンパイル使用メモリ | 204,320 KB |
最終ジャッジ日時 | 2025-01-10 01:59:38 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h>using namespace std;typedef long long ll;#define inf 10e17#define rep(i,n) for(long long i=0; i<n; i++)#define repr(i,n,m) for(long long i=m; i<n; i++)#define mod 1000000007#define sorti(x) sort(x.begin(), x.end())#define sortd(x) sort(x.begin(), x.end(), std::greater<long long>())#define debug(x) std::cerr << (x) << std::endl;#define roll(x) for (auto&& itr : x) { cerr << (itr) << " "; }template <class T> inline void chmax(T &ans, T t) { if (t > ans) ans = t;}template <class T> inline void chmin(T &ans, T t) { if (t < ans) ans = t;}template <typename T>class SegTree {int n; // 葉の数vector<T> data; // データを格納するvectorT def; // 初期値かつ単位元function<T(T, T)> operation; // 区間クエリで使う処理function<T(T, T)> update; // 点更新で使う処理// 区間[a, b)の総和。ノードk=[l, r)に着目している。T _query(int a, int b, int k, int l, int r) {if (r <= a || b <= l) return def; // 交差しないif (a <= l && r <= b)return data[k]; // a,l,r,bの順で完全に含まれるelse {T c1 = _query(a, b, 2 * k + 1, l, (l + r) / 2); // 左の子T c2 = _query(a, b, 2 * k + 2, (l + r) / 2, r); // 右の子return operation(c1, c2);}}public:/*** コンストラクタ* @param _n 必要サイズ* @param _def 初期値かつ単位元* @param _operation クエリ関数* @param _update 更新関数*/SegTree(size_t _n, T _def, function<T(T, T)> _operation, function<T(T, T)> _update): def(_def), operation(_operation), update(_update) {n = 1;while (n < _n) {n *= 2;}data = vector<T>(2 * n - 1, def);}SegTree(vector<T>& A, T _def, function<T(T, T)> _operation, function<T(T, T)> _update): def(_def), operation(_operation), update(_update) {n = 1;auto _n = A.size();while (n < _n) {n *= 2;}data.resize(2 * n, _def);for (int i = 0; i < _n; ++i) data[i + n - 1] = A[i];for (int i = n - 2; i >= 0; --i) data[i] = operation(data[i*2+1], data[i*2+2]);}/*** [a, b)の区間クエリを実行* @param a 左端* @param b 右端* @return クエリの結果*/T query(int a, int b) {return _query(a, b, 0, 0, n);}/*** 場所i(0-indexed)の値をxで更新* @param i* @param x*/void change(int i, T x) {i += n - 1;data[i] = update(data[i], x);while (i > 0) {i = (i - 1) / 2;data[i] = operation(data[i * 2 + 1], data[i * 2 + 2]);}}T operator[](int i) {return data[i + n - 1];}};// 最大公約数を求める(ユークリッドの互除法).long long gcd (long long x, long long y) {if (y > x) swap(x,y);if (y == 0) return x;return gcd(x%y,y);}signed main() {int n;cin >> n;vector<ll> a(n);repr(i, n, 0) {cin >> a[i];}SegTree<ll> segTree(a, 0,[](auto const a, auto const b){ return gcd(a,b); },[](auto const a, auto const b){ return b; });ll ans = 0;ll r = 0;for (ll l = 0; l < n; ++l) {if (l > r) r = l;while (r < n and segTree.query(l, r + 1) > 1) {r ++;}ans += n - r;}cout << ans << endl;}