結果
問題 | No.1300 Sum of Inversions |
ユーザー | naskya |
提出日時 | 2022-01-03 00:32:34 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 306 ms / 2,000 ms |
コード長 | 2,383 bytes |
コンパイル時間 | 1,680 ms |
コンパイル使用メモリ | 110,520 KB |
実行使用メモリ | 24,304 KB |
最終ジャッジ日時 | 2024-04-20 09:56:41 |
合計ジャッジ時間 | 10,254 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 211 ms
19,644 KB |
testcase_04 | AC | 208 ms
19,192 KB |
testcase_05 | AC | 159 ms
16,376 KB |
testcase_06 | AC | 248 ms
21,632 KB |
testcase_07 | AC | 237 ms
20,644 KB |
testcase_08 | AC | 267 ms
22,496 KB |
testcase_09 | AC | 270 ms
22,188 KB |
testcase_10 | AC | 128 ms
14,156 KB |
testcase_11 | AC | 133 ms
14,344 KB |
testcase_12 | AC | 204 ms
19,036 KB |
testcase_13 | AC | 200 ms
18,676 KB |
testcase_14 | AC | 306 ms
24,016 KB |
testcase_15 | AC | 261 ms
22,248 KB |
testcase_16 | AC | 219 ms
19,852 KB |
testcase_17 | AC | 124 ms
13,732 KB |
testcase_18 | AC | 142 ms
15,232 KB |
testcase_19 | AC | 178 ms
17,408 KB |
testcase_20 | AC | 181 ms
17,648 KB |
testcase_21 | AC | 177 ms
17,788 KB |
testcase_22 | AC | 159 ms
16,240 KB |
testcase_23 | AC | 255 ms
21,220 KB |
testcase_24 | AC | 170 ms
16,868 KB |
testcase_25 | AC | 138 ms
15,036 KB |
testcase_26 | AC | 135 ms
14,816 KB |
testcase_27 | AC | 159 ms
16,064 KB |
testcase_28 | AC | 276 ms
22,828 KB |
testcase_29 | AC | 181 ms
17,572 KB |
testcase_30 | AC | 259 ms
22,292 KB |
testcase_31 | AC | 160 ms
16,592 KB |
testcase_32 | AC | 166 ms
17,016 KB |
testcase_33 | AC | 21 ms
7,168 KB |
testcase_34 | AC | 37 ms
7,168 KB |
testcase_35 | AC | 153 ms
24,304 KB |
testcase_36 | AC | 159 ms
24,176 KB |
ソースコード
#include <algorithm> #include <atcoder/modint> #include <map> #include <vector> #include <iostream> #include <iterator> template <class T> struct BIT { int n; std::vector<T> v; BIT(int N):n(N),v(N+1){} void add(int p,T x){for(++p;p<n;p+=p&-p)v[p]+=x;} T psum(int i){T r=0;for(;i>0;i-=i&-i)r+=v[i];return r;} T sum(int l,int r){return psum(r)-psum(l-1);} }; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int N; std::cin >> N; std::vector<int> A(N); std::copy_n(std::istream_iterator<int>(std::cin), N, std::begin(A)); std::map<int, int> count; std::for_each(std::cbegin(A), std::cend(A), [&](const int a) { ++count[a]; }); std::vector<std::pair<int, int>> L, R; std::for_each(std::cbegin(count), std::cend(count), [&](const auto p) { L.emplace_back(p); R.emplace_back(p); }); std::sort(std::begin(L), std::end(L)); std::sort(std::rbegin(R), std::rend(R)); const int M = static_cast<int>(std::size(L)); BIT<long long> L_acc(M), R_acc(M); BIT<int> L_cnt(M), R_cnt(M); for (int i = 0; i < M; i++) { L_acc.add(i, L[i].first * L[i].second); L_cnt.add(i, L[i].second); R_acc.add(i, R[i].first * R[i].second); R_cnt.add(i, R[i].second); } using mint = atcoder::modint998244353; std::vector<mint> right(N), left(N); std::vector<int> rs(N), ls(N); for (int i = 0; i < N - 1; i++) { const int idx = static_cast<int>(std::distance(std::cbegin(L), std::lower_bound(std::cbegin(L), std::cend(L), std::pair {A[i], 0}))); L_cnt.add(idx, -1); L_acc.add(idx, -A[i]); if (i == 0 || idx == 0) continue; right[i] = mint(L_acc.sum(0, idx)); rs[i] = L_cnt.sum(0, idx); } for (int i = 0; i < N - 1; i++) { const int idx = static_cast<int>(std::distance(std::lower_bound(std::crbegin(R), std::crend(R), std::pair {A[N - 1 - i], 0}), std::crend(R)) - 1); --R[idx].second; R_cnt.add(idx, -1); R_acc.add(idx, -A[N - 1 - i]); if (i == 0 || idx == 0) continue; left[i] = mint(R_acc.sum(0, idx)); ls[i] = R_cnt.sum(0, idx); } std::reverse(std::begin(left), std::end(left)); std::reverse(std::begin(ls), std::end(ls)); mint res; for (int i = 0; i < N - 1; i++) { res += mint(right[i]) * ls[i]; res += mint(left[i]) * rs[i]; res += mint(A[i]) * rs[i] * ls[i]; } std::cout << res.val() << '\n'; }