結果

問題 No.1300 Sum of Inversions
ユーザー auauaauaua
提出日時 2020-11-28 14:27:52
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,043 ms / 2,000 ms
コード長 1,999 bytes
コンパイル時間 1,996 ms
コンパイル使用メモリ 184,388 KB
実行使用メモリ 39,296 KB
最終ジャッジ日時 2024-09-12 22:26:38
合計ジャッジ時間 25,508 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 729 ms
30,976 KB
testcase_04 AC 720 ms
30,208 KB
testcase_05 AC 590 ms
25,600 KB
testcase_06 AC 831 ms
34,304 KB
testcase_07 AC 835 ms
32,896 KB
testcase_08 AC 962 ms
35,968 KB
testcase_09 AC 945 ms
35,712 KB
testcase_10 AC 454 ms
21,888 KB
testcase_11 AC 443 ms
22,016 KB
testcase_12 AC 712 ms
30,336 KB
testcase_13 AC 734 ms
29,696 KB
testcase_14 AC 1,043 ms
38,656 KB
testcase_15 AC 962 ms
35,584 KB
testcase_16 AC 781 ms
31,616 KB
testcase_17 AC 395 ms
21,248 KB
testcase_18 AC 517 ms
23,808 KB
testcase_19 AC 596 ms
27,264 KB
testcase_20 AC 615 ms
27,904 KB
testcase_21 AC 652 ms
27,776 KB
testcase_22 AC 559 ms
25,600 KB
testcase_23 AC 814 ms
34,176 KB
testcase_24 AC 599 ms
26,240 KB
testcase_25 AC 498 ms
23,168 KB
testcase_26 AC 490 ms
22,912 KB
testcase_27 AC 572 ms
24,960 KB
testcase_28 AC 936 ms
36,736 KB
testcase_29 AC 673 ms
27,520 KB
testcase_30 AC 965 ms
35,712 KB
testcase_31 AC 594 ms
25,856 KB
testcase_32 AC 618 ms
26,496 KB
testcase_33 AC 47 ms
11,008 KB
testcase_34 AC 102 ms
11,008 KB
testcase_35 AC 509 ms
39,296 KB
testcase_36 AC 530 ms
39,168 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