結果

問題 No.1300 Sum of Inversions
ユーザー auauaauaua
提出日時 2020-11-28 14:29:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 836 ms / 2,000 ms
コード長 2,029 bytes
コンパイル時間 1,952 ms
コンパイル使用メモリ 183,380 KB
実行使用メモリ 39,168 KB
最終ジャッジ日時 2024-09-12 22:26:59
合計ジャッジ時間 20,704 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 563 ms
30,976 KB
testcase_04 AC 545 ms
30,080 KB
testcase_05 AC 441 ms
25,600 KB
testcase_06 AC 615 ms
34,304 KB
testcase_07 AC 649 ms
33,024 KB
testcase_08 AC 730 ms
35,968 KB
testcase_09 AC 732 ms
35,712 KB
testcase_10 AC 330 ms
22,016 KB
testcase_11 AC 315 ms
22,016 KB
testcase_12 AC 524 ms
30,336 KB
testcase_13 AC 573 ms
29,696 KB
testcase_14 AC 836 ms
38,656 KB
testcase_15 AC 758 ms
35,584 KB
testcase_16 AC 630 ms
31,616 KB
testcase_17 AC 318 ms
21,248 KB
testcase_18 AC 424 ms
23,808 KB
testcase_19 AC 481 ms
27,264 KB
testcase_20 AC 508 ms
27,904 KB
testcase_21 AC 530 ms
27,776 KB
testcase_22 AC 465 ms
25,728 KB
testcase_23 AC 665 ms
34,176 KB
testcase_24 AC 486 ms
26,240 KB
testcase_25 AC 388 ms
23,168 KB
testcase_26 AC 391 ms
22,784 KB
testcase_27 AC 459 ms
24,960 KB
testcase_28 AC 739 ms
36,736 KB
testcase_29 AC 505 ms
27,520 KB
testcase_30 AC 772 ms
35,712 KB
testcase_31 AC 467 ms
25,856 KB
testcase_32 AC 462 ms
26,496 KB
testcase_33 AC 44 ms
11,136 KB
testcase_34 AC 101 ms
11,008 KB
testcase_35 AC 348 ms
39,168 KB
testcase_36 AC 367 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){
        ll id=idx[a[i]];
        cntl[i]=bit.sum(now-1)-bit.sum(id);
        bit.add(id,1);
    }
    for(int i=n-1;i>=0;i--){
        ll id=idx[a[i]];
        if(id>0)cntr[i]=bit1.sum(id-1);
        bit1.add(id,1);
    }
    rep(i,n){
        ll id=idx[a[i]];
        suml[i]=bit2.sum(now-1)-bit2.sum(id);
        suml[i]%=mod;
        bit2.add(id,a[i]);
    }
    for(int i=n-1;i>=0;i--){
        ll id=idx[a[i]];
        if(id>0)sumr[i]=bit3.sum(id-1);
        sumr[i]%=mod;
        bit3.add(id,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