結果

問題 No.1300 Sum of Inversions
ユーザー hiro71687khiro71687k
提出日時 2023-03-02 18:52:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,316 ms / 2,000 ms
コード長 1,829 bytes
コンパイル時間 4,820 ms
コンパイル使用メモリ 277,244 KB
実行使用メモリ 43,036 KB
最終ジャッジ日時 2024-09-17 12:53:19
合計ジャッジ時間 36,051 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 963 ms
37,796 KB
testcase_04 AC 939 ms
37,256 KB
testcase_05 AC 731 ms
26,112 KB
testcase_06 AC 1,109 ms
39,976 KB
testcase_07 AC 1,054 ms
39,112 KB
testcase_08 AC 1,170 ms
41,028 KB
testcase_09 AC 1,189 ms
40,780 KB
testcase_10 AC 600 ms
23,680 KB
testcase_11 AC 609 ms
23,808 KB
testcase_12 AC 946 ms
37,292 KB
testcase_13 AC 959 ms
36,892 KB
testcase_14 AC 1,316 ms
42,836 KB
testcase_15 AC 1,191 ms
40,844 KB
testcase_16 AC 988 ms
38,088 KB
testcase_17 AC 571 ms
23,296 KB
testcase_18 AC 692 ms
24,960 KB
testcase_19 AC 833 ms
35,472 KB
testcase_20 AC 850 ms
35,632 KB
testcase_21 AC 837 ms
35,720 KB
testcase_22 AC 750 ms
26,112 KB
testcase_23 AC 1,096 ms
39,900 KB
testcase_24 AC 786 ms
26,624 KB
testcase_25 AC 636 ms
24,448 KB
testcase_26 AC 626 ms
24,192 KB
testcase_27 AC 719 ms
25,728 KB
testcase_28 AC 1,221 ms
41,560 KB
testcase_29 AC 827 ms
35,484 KB
testcase_30 AC 1,196 ms
40,852 KB
testcase_31 AC 759 ms
26,112 KB
testcase_32 AC 795 ms
26,752 KB
testcase_33 AC 50 ms
8,064 KB
testcase_34 AC 105 ms
8,064 KB
testcase_35 AC 678 ms
43,036 KB
testcase_36 AC 697 ms
43,016 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using ld=long double;
ld pie=3.14159265359;
ll inf=10000000000000001;
ll mod=998244353;
template <typename T>
vector<T> compress(vector<T> &X) {
    // ソートした結果を vals に
    vector<T> vals = X;
    sort(vals.begin(), vals.end());
    // 隣り合う重複を削除(unique), 末端のゴミを削除(erase)
    vals.erase(unique(vals.begin(), vals.end()), vals.end());
    // 各要素ごとに二分探索で位置を求める
    for (int i = 0; i < (int)X.size(); i++) {
        X[i] = lower_bound(vals.begin(), vals.end(), X[i]) - vals.begin();
    }
    return vals;
}
ll op(ll a,ll b){
    return (a+b)%mod;
}
ll e(){
    return 0;
}
int main(){
    ll n;
    cin >> n;
    vector<ll>a(n);
    for (ll i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    vector<ll>aa=a;
    vector<ll>b=compress(aa);
    map<ll,ll>memo;
    for (ll i = 0; i < b.size(); i++)
    {
        memo[b[i]]=i;
    }
    vector<ll>c(b.size(),0),d(b.size(),0),ee(b.size(),0),f(b.size(),0);
    for (ll i = 0; i < a.size(); i++)
    {
        c[memo[a[i]]]+=1;
        ee[memo[a[i]]]+=a[i];
        ee[memo[a[i]]]%=mod;
    }
    ll last=b.size();
    segtree<ll,op,e>seg1(c),seg2(d),seg3(ee),seg4(f);
    ll ans=0;
    for (ll i = 0; i < n; i++)
    {
        ll y=memo[a[i]];
        ll syoko=seg1.prod(0,y);
        ll syo=seg3.prod(0,y);
        ll daiko=seg2.prod(y+1,last);
        ll dai=seg4.prod(y+1,last);
        ans+=syoko*dai;
        ans%=mod;
        ans+=daiko*syo;
        ans%=mod;
        ans+=((syoko*daiko)%mod)*a[i];
        seg1.set(y,seg1.get(y)-1);
        seg3.set(y,seg3.get(y)-a[i]);
        seg2.set(y,seg2.get(y)+1);
        seg4.set(y,seg4.get(y)+a[i]);
    }
    cout << ans << endl;
}
0