結果
問題 | No.2821 A[i] ← 2A[j] - A[i] |
ユーザー | Aeren |
提出日時 | 2024-07-26 22:30:49 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 103 ms / 1,500 ms |
コード長 | 10,746 bytes |
コンパイル時間 | 3,675 ms |
コンパイル使用メモリ | 278,268 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-26 22:30:56 |
合計ジャッジ時間 | 6,665 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,812 KB |
testcase_01 | AC | 66 ms
6,940 KB |
testcase_02 | AC | 96 ms
6,940 KB |
testcase_03 | AC | 103 ms
6,944 KB |
testcase_04 | AC | 55 ms
6,940 KB |
testcase_05 | AC | 55 ms
6,940 KB |
testcase_06 | AC | 34 ms
6,940 KB |
testcase_07 | AC | 32 ms
6,944 KB |
testcase_08 | AC | 50 ms
6,944 KB |
testcase_09 | AC | 17 ms
6,940 KB |
testcase_10 | AC | 20 ms
6,940 KB |
testcase_11 | AC | 19 ms
6,940 KB |
testcase_12 | AC | 16 ms
6,940 KB |
testcase_13 | AC | 25 ms
6,940 KB |
testcase_14 | AC | 18 ms
6,944 KB |
testcase_15 | AC | 25 ms
6,940 KB |
testcase_16 | AC | 14 ms
6,944 KB |
testcase_17 | AC | 20 ms
6,944 KB |
testcase_18 | AC | 23 ms
6,940 KB |
testcase_19 | AC | 4 ms
6,944 KB |
testcase_20 | AC | 5 ms
6,940 KB |
testcase_21 | AC | 2 ms
6,940 KB |
testcase_22 | AC | 3 ms
6,940 KB |
testcase_23 | AC | 6 ms
6,940 KB |
testcase_24 | AC | 2 ms
6,940 KB |
testcase_25 | AC | 2 ms
6,944 KB |
testcase_26 | AC | 2 ms
6,940 KB |
testcase_27 | AC | 2 ms
6,944 KB |
testcase_28 | AC | 2 ms
6,940 KB |
testcase_29 | AC | 2 ms
6,944 KB |
testcase_30 | AC | 2 ms
6,940 KB |
testcase_31 | AC | 2 ms
6,940 KB |
testcase_32 | AC | 3 ms
6,940 KB |
testcase_33 | AC | 2 ms
6,940 KB |
ソースコード
// #include <bits/allocator.h> // Temp fix for gcc13 global pragma // #pragma GCC target("avx2,bmi2,popcnt,lzcnt") // #pragma GCC optimize("O3,unroll-loops") #include <bits/stdc++.h> // #include <x86intrin.h> using namespace std; #if __cplusplus >= 202002L using namespace numbers; #endif #ifdef LOCAL #include "Debug.h" #else #define debug_endl() 42 #define debug(...) 42 #define debug2(...) 42 #define debugbin(...) 42 #endif template<bool HAS_QUERY, bool HAS_UPDATE, class T, class U, class F1, class F2, class F3> struct segment_tree_base{ static_assert(HAS_QUERY || HAS_UPDATE); #define ifQ if constexpr(HAS_QUERY) #define ifU if constexpr(HAS_UPDATE) int n, size, log; vector<T> data; vector<U> data_action; F1 TT; // monoid operation (always adjacent) T T_id; // monoid identity F2 UU; // monoid operation (superset, subset) U U_id; // monoid identity F3 UT; // action of U on T (superset, subset) // O(n) segment_tree_base(F1 TT, T T_id, F2 UU, U U_id, F3 UT): TT(TT), T_id(T_id), UU(UU), U_id(U_id), UT(UT){ } segment_tree_base &operator=(const segment_tree_base &seg){ n = seg.n; size = seg.size; log = seg.log; data = seg.data; data_action = seg.data_action; } // O(1) friend void swap(segment_tree_base &x, segment_tree_base &y){ swap(x.n, y.n); swap(x.size, y.size); swap(x.log, y.log); swap(x.data, y.data); swap(x.data_action, y.data_action); } // O(n) void build(int n){ assert(n >= 0); this->n = n; size = 1; while(size < n) size <<= 1; log = __lg(size); ifQ data.assign(size << 1, T_id); ifU data_action.assign(HAS_QUERY ? size : size << 1, U_id); } // O(n) void build(int n, T x){ static_assert(HAS_QUERY); assert(n >= 0); this->n = n; size = 1; while(size < n) size <<= 1; log = __lg(size); data.assign(size << 1, T_id); fill(data.begin() + size, data.begin() + size + n, x); for(auto i = size - 1; i >= 1; -- i) refresh(i); ifU data_action.assign(size, U_id); } // O(n) template<class V> void build(const vector<V> &a){ static_assert(HAS_QUERY); n = (int)a.size(); size = 1; while(size < n) size <<= 1; log = __lg(size); data.assign(size << 1, T_id); copy(a.begin(), a.end(), data.begin() + size); for(auto i = size - 1; i >= 1; -- i) refresh(i); ifU data_action.assign(size, U_id); } // O(n) void build_action(int n){ static_assert(!HAS_QUERY && HAS_UPDATE); assert(n >= 0); build(n); } // O(n) void build_action(int n, U f){ static_assert(!HAS_QUERY && HAS_UPDATE); assert(n >= 0); this->n = n; size = 1; while(size < n) size <<= 1; log = __lg(size); data_action.assign(size << 1, U_id); fill(data_action.begin() + size, data_action.begin() + size + n, f); } // O(n) template<class V> void build_action(const vector<V> &a){ static_assert(!HAS_QUERY && HAS_UPDATE); n = (int)a.size(); size = 1; while(size < n) size <<= 1; log = __lg(size); data_action.assign(size << 1, U_id); copy(a.begin(), a.end(), data_action.begin() + size); } // O(1) void refresh(int u){ static_assert(HAS_QUERY); data[u] = TT(data[u << 1], data[u << 1 | 1]); } // O(1) void apply(int u, U f){ static_assert(HAS_UPDATE); ifQ data[u] = UT(f, data[u]); if(!HAS_QUERY || u < size) data_action[u] = UU(f, data_action[u]); } // O(1) void push(int u){ static_assert(HAS_UPDATE); apply(u << 1, data_action[u]), apply(u << 1 | 1, data_action[u]); data_action[u] = U_id; } // O(log(n)) if HAS_UPDATE, O(1) otherwise. T query(int p){ static_assert(HAS_QUERY); assert(0 <= p && p < n); p += size; ifU for(auto i = log; i >= 1; -- i) push(p >> i); return data[p]; } // O(log(n)) U query_action(int p){ static_assert(!HAS_QUERY && HAS_UPDATE); assert(0 <= p && p < n); p += size; ifU for(auto i = log; i >= 1; -- i) push(p >> i); return data_action[p]; } // O(log(n)) T query(int ql, int qr){ static_assert(HAS_QUERY); assert(0 <= ql && ql <= qr && qr <= n); if(ql == qr) return T_id; ql += size, qr += size; ifU for(auto i = log; i >= 1; -- i){ if(ql >> i << i != ql) push(ql >> i); if(qr >> i << i != qr) push(qr >> i); } T res_left = T_id, res_right = T_id; for(; ql < qr; ql >>= 1, qr >>= 1){ if(ql & 1) res_left = TT(res_left, data[ql ++]); if(qr & 1) res_right = TT(data[-- qr], res_right); } return TT(res_left, res_right); } // O(1) T query_all() const{ static_assert(HAS_QUERY); return data[1]; } // O(n) vector<T> to_array(){ static_assert(HAS_QUERY); ifU for(auto u = 1; u < size; ++ u) push(u); return vector<T>(data.begin() + size, data.begin() + size + n); } // O(n) vector<U> to_array_of_updates(){ static_assert(!HAS_QUERY && HAS_UPDATE); for(auto u = 1; u < size; ++ u) push(u); return vector<U>(data_action.begin() + size, data_action.begin() + size + n); } // O(log(n)) void set(int p, T x){ static_assert(HAS_QUERY); assert(0 <= p && p < n); p += size; ifU for(auto i = log; i >= 1; -- i) push(p >> i); data[p] = x; for(auto i = 1; i <= log; ++ i) refresh(p >> i); } // O(log(n)) void set_action(int p, U f){ static_assert(!HAS_QUERY && HAS_UPDATE); assert(0 <= p && p < n); p += size; for(auto i = log; i >= 1; -- i) push(p >> i); data_action[p] = f; } // O(log(n)) void update(int p, U f){ static_assert(HAS_UPDATE); assert(0 <= p && p < n); p += size; for(auto i = log; i >= 1; -- i) push(p >> i); ifQ{ data[p] = UT(f, data[p]); for(auto i = 1; i <= log; ++ i) refresh(p >> i); } else data_action[p] = UU(f, data_action[p]); } // O(log(n)) void update(int ql, int qr, U f){ static_assert(HAS_UPDATE); assert(0 <= ql && ql <= qr && qr <= n); if(ql == qr) return; ql += size, qr += size; for(auto i = log; i >= 1; -- i){ if(ql >> i << i != ql) push(ql >> i); if(qr >> i << i != qr) push(qr >> i); } int _ql = ql, _qr = qr; for(; ql < qr; ql >>= 1, qr >>= 1){ if(ql & 1) apply(ql ++, f); if(qr & 1) apply(-- qr, f); } ql = _ql, qr = _qr; ifQ for(auto i = 1; i <= log; ++ i){ if(ql >> i << i != ql) refresh(ql >> i); if(qr >> i << i != qr) refresh(qr >> i); } } void update_beats(int ql, int qr, auto exit_rule, auto enter_rule, auto update_rule){ static_assert(HAS_QUERY && HAS_UPDATE); assert(0 <= ql && ql <= qr && qr <= n); if(ql == qr) return; ql += size, qr += size; for(auto i = log; i >= 1; -- i){ if(ql >> i << i != ql) push(ql >> i); if(qr >> i << i != qr) push(qr >> i); } auto recurse = [&](auto self, int u)->void{ if(exit_rule(data[u])) return; if(enter_rule(data[u])){ apply(u, update_rule(data[u])); return; } push(u); self(self, u << 1), self(self, u << 1 | 1); refresh(u); }; int _ql = ql, _qr = qr; for(; ql < qr; ql >>= 1, qr >>= 1){ if(ql & 1) recurse(recurse, ql ++); if(qr & 1) recurse(recurse, -- qr); } ql = _ql, qr = _qr; for(auto i = 1; i <= log; ++ i){ if(ql >> i << i != ql) refresh(ql >> i); if(qr >> i << i != qr) refresh(qr >> i); } } // pred(sum[ql, r)) is T, T, ..., T, F, F, ..., F // Returns max r with T // O(log(n)) int max_pref(int ql, auto pred){ static_assert(HAS_QUERY); assert(0 <= ql && ql <= n && pred(T_id)); if(ql == n) return n; ql += size; ifU for(auto i = log; i >= 1; -- i) push(ql >> i); T sum = T_id; do{ while(~ql & 1) ql >>= 1; if(!pred(TT(sum, data[ql]))){ while(ql < size){ ifU push(ql); ql = ql << 1; if(pred(TT(sum, data[ql]))) sum = TT(sum, data[ql ++]); } return ql - size; } sum = TT(sum, data[ql]); ++ ql; }while((ql & -ql) != ql); return n; } // pred(sum[l, qr)) is F, F, ..., F, T, T, ..., T // Returns min l with T // O(log(n)) int max_suff(int qr, auto pred){ static_assert(HAS_QUERY); assert(0 <= qr && qr <= n && pred(T_id)); if(qr == 0) return 0; qr += size; ifU for(auto i = log; i >= 1; -- i) push(qr - 1 >> i); T sum = T_id; do{ -- qr; while(qr > 1 && qr & 1) qr >>= 1; if(!pred(TT(data[qr], sum))){ while(qr < size){ ifU push(qr); qr = qr << 1 | 1; if(pred(TT(data[qr], sum))) sum = TT(data[qr --], sum); } return qr + 1 - size; } sum = TT(data[qr], sum); }while((qr & -qr) != qr); return 0; } template<class output_stream> friend output_stream &operator<<(output_stream &out, segment_tree_base<HAS_QUERY, HAS_UPDATE, T, U, F1, F2, F3> seg){ out << "{"; for(auto i = 0; i < seg.n; ++ i){ ifQ out << seg.query(i); else out << seg.query_action(i); if(i != seg.n - 1) out << ", "; } return out << '}'; } #undef ifQ #undef ifU }; // Supports query template<class T, class F> auto make_Q_segment_tree(F TT, T T_id){ using U = int; auto _UU = [&](U, U)->U{ return U{}; }; auto _UT = [&](U, T)->T{ return T{}; }; return segment_tree_base<true, false, T, U, F, decltype(_UU), decltype(_UT)>(TT, T_id, _UU, U{}, _UT); } // Supports update template<class U, class F> auto make_U_segment_tree(F UU, U U_id){ using T = int; auto _TT = [&](T, T)->T{ return T{}; }; auto _UT = [&](U, T)->T{ return T{}; }; return segment_tree_base<false, true, T, U, decltype(_TT), F, decltype(_UT)>(_TT, T{}, UU, U_id, _UT); } // Supports query and update template<class T, class U, class F1, class F2, class F3> auto make_QU_segment_tree(F1 TT, T T_id, F2 UU, U U_id, F3 UT){ return segment_tree_base<true, true, T, U, F1, F2, F3>(TT, T_id, UU, U_id, UT); } int main(){ cin.tie(0)->sync_with_stdio(0); cin.exceptions(ios::badbit | ios::failbit); auto seg = make_Q_segment_tree([&](long long x, long long y){ return gcd(x, y); }, 0LL); int n; cin >> n; vector<long long> a(n); copy_n(istream_iterator<long long>(cin), n, a.begin()); vector<long long> cmpr; { set<long long> s; for(auto x: a){ if(s.contains(x)){ continue; } auto it = s.lower_bound(x); if(it != s.begin()){ cmpr.push_back(x - *prev(it)); } if(it != s.end()){ cmpr.push_back(*it - x); } s.insert(it, x); } } ranges::sort(cmpr); cmpr.erase(ranges::unique(cmpr).begin(), cmpr.end()); auto ind = [&](long long x)->int{ return ranges::lower_bound(cmpr, x) - cmpr.begin(); }; seg.build((int)cmpr.size()); { set<long long> s; map<long long, int> cnt; for(auto x: a){ if(!s.contains(x)){ auto it = s.lower_bound(x); if(it != s.begin() && it != s.end()){ long long dif = *it - *prev(it); if(-- cnt[dif] == 0){ seg.set(ind(dif), seg.T_id); } } if(it != s.begin()){ long long dif = x - *prev(it); ++ cnt[dif]; seg.set(ind(dif), dif); } if(it != s.end()){ long long dif = *it - x; ++ cnt[dif]; seg.set(ind(dif), dif); } s.insert(x); } cout << seg.query_all() << "\n"; } } return 0; } /* */