結果
問題 | No.1733 Sum of Sorted Subarrays |
ユーザー |
![]() |
提出日時 | 2021-10-20 14:20:18 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 284 ms / 3,000 ms |
コード長 | 1,121 bytes |
コンパイル時間 | 4,317 ms |
コンパイル使用メモリ | 262,116 KB |
最終ジャッジ日時 | 2025-01-25 01:59:17 |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; using mint = modint998244353; #define pb(...) emplace_back(__VA_ARGS__) #define all(x) x.begin(), x.end() #define rep(i, n) for (int i = 0; i < (n); i++) using pii = pair<mint, mint>; pii lmerge(pii a, pii b) { return pii(a.first * b.second + b.first, a.second * b.second); } pii rmerge(pii a, pii b) { return pii(a.first + b.first * a.second, a.second * b.second); } pii e(){ return pii(0, 1); } int solve(int n, vector<int> a) { vector<pair<int, int>> v; rep(i, n) v.pb(a[i], i); sort(all(v)); segtree<pii, lmerge, e> segl(n); segtree<pii, rmerge, e> segr(n); rep(i, n) segl.set(i, {1, 1}), segr.set(i, {1, 1}); mint ans = 0; rep(i, n) { int na = v[i].first, idx = v[i].second; ans += segl.prod(0, idx + 1).first * segr.prod(idx, n).first * na; segl.set(idx, {2, 2}); segr.set(idx, {2, 2}); } return ans.val(); } int main() { int n; cin >> n; vector<int> a(n); rep(i, n) cin >> a[i]; cout << solve(n, a) << endl; }