結果

問題 No.1300 Sum of Inversions
ユーザー auauaauaua
提出日時 2020-11-28 14:27:52
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,099 ms / 2,000 ms
コード長 1,999 bytes
コンパイル時間 1,791 ms
コンパイル使用メモリ 180,936 KB
実行使用メモリ 38,940 KB
最終ジャッジ日時 2023-10-09 23:58:30
合計ジャッジ時間 26,682 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,348 KB
testcase_01 AC 2 ms
4,356 KB
testcase_02 AC 2 ms
4,352 KB
testcase_03 AC 755 ms
30,740 KB
testcase_04 AC 757 ms
29,912 KB
testcase_05 AC 604 ms
25,404 KB
testcase_06 AC 855 ms
34,180 KB
testcase_07 AC 890 ms
32,928 KB
testcase_08 AC 991 ms
35,704 KB
testcase_09 AC 955 ms
35,448 KB
testcase_10 AC 455 ms
21,832 KB
testcase_11 AC 444 ms
21,772 KB
testcase_12 AC 747 ms
30,180 KB
testcase_13 AC 768 ms
29,480 KB
testcase_14 AC 1,099 ms
38,424 KB
testcase_15 AC 976 ms
35,184 KB
testcase_16 AC 842 ms
31,260 KB
testcase_17 AC 430 ms
20,940 KB
testcase_18 AC 596 ms
23,580 KB
testcase_19 AC 637 ms
26,996 KB
testcase_20 AC 652 ms
27,576 KB
testcase_21 AC 689 ms
27,524 KB
testcase_22 AC 601 ms
25,580 KB
testcase_23 AC 878 ms
34,152 KB
testcase_24 AC 623 ms
26,020 KB
testcase_25 AC 519 ms
22,904 KB
testcase_26 AC 517 ms
22,612 KB
testcase_27 AC 589 ms
24,664 KB
testcase_28 AC 945 ms
36,688 KB
testcase_29 AC 690 ms
27,312 KB
testcase_30 AC 973 ms
35,496 KB
testcase_31 AC 631 ms
25,464 KB
testcase_32 AC 632 ms
26,304 KB
testcase_33 AC 45 ms
11,084 KB
testcase_34 AC 98 ms
10,936 KB
testcase_35 AC 514 ms
38,932 KB
testcase_36 AC 532 ms
38,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
//#define int long long
#define REP(i,m,n) for(int i=(m);i<(n);i++)
#define rep(i,n) REP(i,0,n)
#define pb push_back
#define all(a) a.begin(),a.end()
#define rall(c) (c).rbegin(),(c).rend()
#define mp make_pair
#define endl '\n'
#define vec vector<ll>
#define mat vector<vector<ll> >
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> pll;
typedef long double ld;
typedef complex<double> comp;
const ll INF=1e9+7;
const ll inf=INF*INF;
const ll MOD=998244353;
const ll mod=MOD;
const ll MAX=200010;
const double PI=acos(-1.0);

//BIT閉区間
template< class T >
struct BIT
{
  vector< T > data;

  BIT(int sz)
  {
    data.assign(++sz, 0);
  }

  T sum(int k)
  {
    T ret = 0;
    for(++k; k > 0; k -= k & -k) ret += data[k];
    return (ret);
  }

  void add(int k, T x)
  {
    for(++k; k < data.size(); k += k & -k) data[k] += x;
  }
};

signed main(){
    ll n;cin>>n;
    vector<ll>a(n);
    set<ll>st;
    rep(i,n){
        cin>>a[i];
        st.insert(a[i]);
    }
    ll now=0;
    map<ll,ll>idx;
    for(auto e:st){
        idx[e]=now;
        now++;
    }
    BIT<ll>bit(now);
    BIT<ll>bit1(now);
    BIT<ll>bit2(now);
    BIT<ll>bit3(now);
    ll ans=0;
    vector<ll>cntl(n),cntr(n),suml(n),sumr(n);
    rep(i,n){
        cntl[i]=bit.sum(now-1)-bit.sum(idx[a[i]]);
        bit.add(idx[a[i]],1);
    }
    for(int i=n-1;i>=0;i--){
        if(idx[a[i]]>0)cntr[i]=bit1.sum(idx[a[i]]-1);
        bit1.add(idx[a[i]],1);
    }
    rep(i,n){
        suml[i]=bit2.sum(now-1)-bit2.sum(idx[a[i]]);
        suml[i]%=mod;
        bit2.add(idx[a[i]],a[i]);
    }
    for(int i=n-1;i>=0;i--){
        if(idx[a[i]]>0)sumr[i]=bit3.sum(idx[a[i]]-1);
        sumr[i]%=mod;
        bit3.add(idx[a[i]],a[i]);
    }
    rep(i,n){
        ans=(ans+a[i]%mod*cntl[i]%mod*cntr[i]%mod)%mod;
        ans=(ans+suml[i]*cntr[i]%mod)%mod;
        ans=(ans+sumr[i]*cntl[i]%mod)%mod;
    }
    cout<<ans<<endl;
}
0