結果
問題 | No.1300 Sum of Inversions |
ユーザー | rodea |
提出日時 | 2020-12-06 16:53:29 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 898 ms / 2,000 ms |
コード長 | 3,035 bytes |
コンパイル時間 | 1,688 ms |
コンパイル使用メモリ | 132,708 KB |
実行使用メモリ | 45,588 KB |
最終ジャッジ日時 | 2023-10-17 15:26:49 |
合計ジャッジ時間 | 23,944 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,348 KB |
testcase_01 | AC | 2 ms
4,348 KB |
testcase_02 | AC | 2 ms
4,348 KB |
testcase_03 | AC | 668 ms
39,532 KB |
testcase_04 | AC | 645 ms
39,100 KB |
testcase_05 | AC | 508 ms
27,664 KB |
testcase_06 | AC | 785 ms
41,856 KB |
testcase_07 | AC | 746 ms
40,876 KB |
testcase_08 | AC | 843 ms
43,048 KB |
testcase_09 | AC | 827 ms
42,948 KB |
testcase_10 | AC | 400 ms
24,816 KB |
testcase_11 | AC | 393 ms
24,928 KB |
testcase_12 | AC | 643 ms
39,196 KB |
testcase_13 | AC | 636 ms
38,580 KB |
testcase_14 | AC | 898 ms
45,024 KB |
testcase_15 | AC | 813 ms
42,816 KB |
testcase_16 | AC | 716 ms
40,144 KB |
testcase_17 | AC | 381 ms
24,496 KB |
testcase_18 | AC | 456 ms
26,164 KB |
testcase_19 | AC | 558 ms
37,036 KB |
testcase_20 | AC | 578 ms
37,328 KB |
testcase_21 | AC | 580 ms
37,276 KB |
testcase_22 | AC | 505 ms
27,648 KB |
testcase_23 | AC | 770 ms
41,808 KB |
testcase_24 | AC | 531 ms
28,032 KB |
testcase_25 | AC | 441 ms
25,784 KB |
testcase_26 | AC | 429 ms
25,612 KB |
testcase_27 | AC | 500 ms
27,044 KB |
testcase_28 | AC | 863 ms
43,712 KB |
testcase_29 | AC | 573 ms
37,172 KB |
testcase_30 | AC | 848 ms
42,908 KB |
testcase_31 | AC | 510 ms
27,756 KB |
testcase_32 | AC | 543 ms
28,160 KB |
testcase_33 | AC | 153 ms
26,836 KB |
testcase_34 | AC | 163 ms
26,836 KB |
testcase_35 | AC | 500 ms
45,588 KB |
testcase_36 | AC | 505 ms
45,588 KB |
ソースコード
#pragma GCC optimize("O3") #include <iostream> #include <iomanip> #include <cstdio> #include <string> #include <cstring> #include <deque> #include <list> #include <queue> #include <stack> #include <vector> #include <utility> #include <algorithm> #include <map> #include <set> #include <complex> #include <cmath> #include <limits> #include <cfloat> #include <climits> #include <ctime> #include <cassert> #include <numeric> #include <fstream> #include <functional> #include <bitset> using namespace std; using ll = long long; using P = pair<int, int>; using T = tuple<int, int, int>; template <class T> inline T chmax(T &a, const T b) {return a = (a < b) ? b : a;} template <class T> inline T chmin(T &a, const T b) {return a = (a > b) ? b : a;} constexpr int mod = 998244353; constexpr int MOD = 1e9 + 7; constexpr int inf = 1e9; constexpr long long INF = 1e18; #define all(a) (a).begin(), (a).end() int dx[] = {1, 0, -1, 0}; int dy[] = {0, 1, 0, -1}; // Range Sum Query class SegmentTree{ public: int n; vector<ll> node; SegmentTree(vector<ll> v) { int sz = v.size(); n = 1; while(n < sz) n *= 2; node.resize(2*n-1, 0); for(int i=0; i<sz; i++) node[i+n-1] = v[i]; for(int i=n-2; i>=0; i--) node[i] = node[i*2+1] + node[i*2+2]; } void add(int x, ll val){ x += (n - 1); node[x] += val; while(0 < x){ x = (x - 1) / 2; node[x] = node[2*x+1] + node[2*x+2]; } } ll getsum(int a, int b, int k=0, int l=0, int r=-1){ if(r < 0) r = n; if(b <= l || r <= a) return 0; if(a <= l && r <= b) return node[k]; ll vl = getsum(a, b, 2*k+1, l, (l+r)/2); ll vr = getsum(a, b, 2*k+2, (l+r)/2, r); return vl + vr; } }; int main(){ cin.tie(0); ios::sync_with_stdio(false); int n; cin>>n; vector<int> a(n); for(int i=0; i<n; i++) cin>>a[i]; set<int> st; for(int i=0; i<n; i++) st.emplace(a[i]); map<int, int> num2idx; int idx = 0; for(auto i:st){ num2idx[i] = idx++; } SegmentTree clseg(vector<ll>(n, 0)), slseg(vector<ll>(n, 0)); vector<ll> cntl(n, 0), suml(n, 0); for(int i=0; i<n; i++){ cntl[i] = clseg.getsum(num2idx[a[i]] + 1, n); suml[i] = slseg.getsum(num2idx[a[i]] + 1, n); clseg.add(num2idx[a[i]], 1); slseg.add(num2idx[a[i]], a[i]); } SegmentTree crseg(vector<ll>(n, 0)), srseg(vector<ll>(n, 0)); vector<ll> cntr(n, 0), sumr(n, 0); for(int i=n-1; i>=0; i--){ cntr[i] = crseg.getsum(0, num2idx[a[i]]); sumr[i] = srseg.getsum(0, num2idx[a[i]]); crseg.add(num2idx[a[i]], 1); srseg.add(num2idx[a[i]], a[i]); } ll ans = 0; for(int i=1; i+1<n; i++){ ll L = (suml[i] % mod) * cntr[i] % mod; ll R = (sumr[i] % mod) * cntl[i] % mod; ll mid = a[i] * cntl[i] % mod * cntr[i] % mod; ans += (L + mid + R) % mod; ans %= mod; } cout << ans << endl; return 0; }