結果
問題 | No.1300 Sum of Inversions |
ユーザー | chocorusk |
提出日時 | 2020-11-27 22:37:38 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 278 ms / 2,000 ms |
コード長 | 2,571 bytes |
コンパイル時間 | 2,526 ms |
コンパイル使用メモリ | 137,944 KB |
実行使用メモリ | 13,920 KB |
最終ジャッジ日時 | 2023-10-09 20:41:22 |
合計ジャッジ時間 | 9,332 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 197 ms
11,424 KB |
testcase_04 | AC | 189 ms
11,200 KB |
testcase_05 | AC | 156 ms
9,872 KB |
testcase_06 | AC | 223 ms
12,476 KB |
testcase_07 | AC | 214 ms
11,944 KB |
testcase_08 | AC | 247 ms
13,068 KB |
testcase_09 | AC | 255 ms
13,064 KB |
testcase_10 | AC | 118 ms
9,048 KB |
testcase_11 | AC | 120 ms
8,884 KB |
testcase_12 | AC | 193 ms
11,288 KB |
testcase_13 | AC | 193 ms
11,224 KB |
testcase_14 | AC | 278 ms
13,920 KB |
testcase_15 | AC | 242 ms
12,940 KB |
testcase_16 | AC | 197 ms
11,732 KB |
testcase_17 | AC | 116 ms
8,576 KB |
testcase_18 | AC | 140 ms
9,396 KB |
testcase_19 | AC | 164 ms
10,456 KB |
testcase_20 | AC | 176 ms
10,452 KB |
testcase_21 | AC | 170 ms
10,500 KB |
testcase_22 | AC | 148 ms
10,032 KB |
testcase_23 | AC | 222 ms
12,516 KB |
testcase_24 | AC | 161 ms
10,152 KB |
testcase_25 | AC | 135 ms
9,128 KB |
testcase_26 | AC | 127 ms
9,104 KB |
testcase_27 | AC | 146 ms
9,608 KB |
testcase_28 | AC | 255 ms
13,348 KB |
testcase_29 | AC | 162 ms
10,556 KB |
testcase_30 | AC | 226 ms
13,004 KB |
testcase_31 | AC | 146 ms
9,836 KB |
testcase_32 | AC | 159 ms
10,372 KB |
testcase_33 | AC | 42 ms
6,200 KB |
testcase_34 | AC | 96 ms
6,240 KB |
testcase_35 | AC | 147 ms
13,900 KB |
testcase_36 | AC | 168 ms
13,892 KB |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #define popcount __builtin_popcount using namespace std; typedef long long ll; typedef pair<int, int> P; const ll MOD=998244353; ll powmod(ll a, ll k){ ll ap=a, ans=1; while(k){ if(k&1){ ans*=ap; ans%=MOD; } ap=ap*ap; ap%=MOD; k>>=1; } return ans; } ll inv(ll a){ return powmod(a, MOD-2); } ll f[2000010], invf[2000010]; void fac(int n){ f[0]=1; for(ll i=1; i<=n; i++) f[i]=f[i-1]*i%MOD; invf[n]=inv(f[n]); for(ll i=n-1; i>=0; i--) invf[i]=invf[i+1]*(i+1)%MOD; } ll comb(int x, int y){ if(!(0<=y && y<=x)) return 0; return f[x]*invf[y]%MOD*invf[x-y]%MOD; } template<typename T> struct BIT{ vector<T> bit; int size; BIT(int n):size(n), bit(n+1, 0){} T sum(int i){ //[0, i) T s=0; while(i>0){ s+=bit[i]; s%=MOD; i-=(i&(-i)); } return s; } T sum(int l, int r){ //[l, r) return (sum(r)-sum(l)+MOD)%MOD; } void add(int i, T x){ i++; while(i<=size){ bit[i]+=x; bit[i]%=MOD; i+=(i&(-i)); } } }; int main() { int n; cin>>n; ll a[200020]; vector<ll> v(n); for(int i=0; i<n; i++){ cin>>a[i]; //a[i]%=MOD; v[i]=a[i]; } sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); int m=v.size(); for(int i=0; i<n; i++){ a[i]=lower_bound(v.begin(), v.end(), a[i])-v.begin(); } BIT<ll> bit(m), bit2(m), bit3(m), bit4(m), bit5(m); ll ans=0; for(int i=0; i<n; i++) bit2.add(a[i], 1); for(int i=0; i<n; i++){ bit2.add(a[i], MOD-1); ll s=bit.sum(a[i]+1, m); (ans+=v[a[i]]%MOD*s%MOD*bit2.sum(a[i]))%=MOD; (ans+=v[a[i]]%MOD*bit3.sum(a[i]+1, m))%=MOD; bit.add(a[i], 1); bit3.add(a[i], s); } reverse(a, a+n); for(int i=0; i<n; i++){ ll s=bit4.sum(a[i]); (ans+=v[a[i]]%MOD*bit5.sum(a[i]))%=MOD; bit4.add(a[i], 1); bit5.add(a[i], s); } cout<<ans<<endl; return 0; }