結果
問題 | No.1300 Sum of Inversions |
ユーザー |
|
提出日時 | 2024-04-01 02:40:11 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 463 ms / 2,000 ms |
コード長 | 886 bytes |
コンパイル時間 | 2,133 ms |
コンパイル使用メモリ | 211,604 KB |
最終ジャッジ日時 | 2025-02-20 18:42:13 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/fenwicktree>#include <atcoder/modint>using namespace atcoder;using namespace std;using ll = long long;using mint = modint998244353;int main() {int N;cin >> N;set<ll> value;vector<ll> A(N + 1, 0);for (int i = 1;i <= N;i++) {cin >> A[i];value.insert(A[i]);}map<ll, int> tos;int K = 0;for (auto v : value) {tos[v] = K +1;K++;}fenwick_tree<mint> bits(K + 10);fenwick_tree<mint> bits2(K + 10);fenwick_tree<ll> cnt(K + 10);fenwick_tree<mint> V(K + 100);vector<mint> T(N + 1, 0);mint ans = 0;for (int i = N;i >= 1;i--) {int now = tos[A[i]];T[i] = (A[i] * cnt.sum(0, now)) + bits.sum(0, now);bits.add(now, A[i]);bits2.add(now, T[i]);cnt.add(now, 1);mint nd = V.sum(0, now);ans += (A[i] * nd) + bits2.sum(0, now);V.add(now, cnt.sum(0, now));}cout << ans.val() << endl;}