結果
問題 | No.742 にゃんにゃんにゃん 猫の挨拶 |
ユーザー |
|
提出日時 | 2020-08-01 23:55:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 14 ms / 2,500 ms |
コード長 | 2,303 bytes |
コンパイル時間 | 1,247 ms |
コンパイル使用メモリ | 87,392 KB |
最終ジャッジ日時 | 2025-01-12 13:00:46 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 16 |
ソースコード
#include <iostream>#include <vector>#include <functional>template <class T>struct SegmentTree {using Merger = std::function<T(T, T)>;int length;std::vector<T> dat;T unit;Merger merge;SegmentTree() = default;SegmentTree(int n, const T& unit, const Merger& merge): length(1), unit(unit), merge(merge) {while (length < n) length <<= 1;dat.assign(length * 2, unit);}template <class Container>SegmentTree(const Container& elems, const T& unit, const Merger& merge): length(1), unit(unit), merge(merge) {int n = elems.size();while (length < n) length <<= 1;dat.assign(length * 2, unit);std::copy(elems.begin(), elems.end(), dat.begin() + length);for (int nidx = length - 1; nidx >= 1; --nidx) {T vl = dat[nidx * 2 + 0];T vr = dat[nidx * 2 + 1];dat[nidx] = merge(vl, vr);}}void update(int nidx, const T& elem) {nidx += length;dat[nidx] = elem;while (nidx > 0) {nidx >>= 1;T vl = dat[nidx * 2 + 0];T vr = dat[nidx * 2 + 1];dat[nidx] = merge(vl, vr);}}T fold(int ql, int qr) const {ql = std::max(ql, 0);qr = std::min(qr, length);ql += length, qr += length;T lacc = unit, racc = unit;while (ql < qr) {if (ql & 1) {lacc = merge(lacc, dat[ql]);++ql;}if (qr & 1) {--qr;racc = merge(dat[qr], racc);}ql >>= 1, qr >>= 1;}return merge(lacc, racc);}T get(int idx) const { return dat[idx + length]; }T fold_all() const { return dat[1]; }};using lint = long long;void solve() {int n;std::cin >> n;SegmentTree<int>seg(n, 0, [](auto a, auto b) { return a + b; });lint ans = 0;for (int i = 0; i < n; ++i) {int x;std::cin >> x;--x;ans += seg.fold(x + 1, n);seg.update(x, 1);}std::cout << ans << "\n";}int main() {std::cin.tie(nullptr);std::ios::sync_with_stdio(false);solve();return 0;}