結果

問題 No.1300 Sum of Inversions
ユーザー monnumonnu
提出日時 2021-07-11 15:44:50
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 222 ms / 2,000 ms
コード長 1,876 bytes
コンパイル時間 1,862 ms
コンパイル使用メモリ 180,000 KB
実行使用メモリ 18,632 KB
最終ジャッジ日時 2023-09-14 20:12:14
合計ジャッジ時間 10,820 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 167 ms
15,188 KB
testcase_04 AC 165 ms
14,724 KB
testcase_05 AC 134 ms
12,768 KB
testcase_06 AC 192 ms
16,456 KB
testcase_07 AC 184 ms
15,980 KB
testcase_08 AC 206 ms
17,496 KB
testcase_09 AC 202 ms
17,428 KB
testcase_10 AC 110 ms
11,244 KB
testcase_11 AC 111 ms
11,188 KB
testcase_12 AC 167 ms
14,908 KB
testcase_13 AC 164 ms
14,788 KB
testcase_14 AC 222 ms
18,632 KB
testcase_15 AC 201 ms
17,120 KB
testcase_16 AC 172 ms
15,460 KB
testcase_17 AC 105 ms
10,912 KB
testcase_18 AC 124 ms
11,972 KB
testcase_19 AC 145 ms
13,744 KB
testcase_20 AC 148 ms
13,892 KB
testcase_21 AC 145 ms
13,588 KB
testcase_22 AC 133 ms
12,872 KB
testcase_23 AC 191 ms
16,596 KB
testcase_24 AC 139 ms
13,072 KB
testcase_25 AC 118 ms
11,764 KB
testcase_26 AC 116 ms
11,476 KB
testcase_27 AC 130 ms
12,556 KB
testcase_28 AC 210 ms
17,716 KB
testcase_29 AC 147 ms
13,592 KB
testcase_30 AC 202 ms
17,100 KB
testcase_31 AC 135 ms
12,940 KB
testcase_32 AC 141 ms
13,224 KB
testcase_33 AC 64 ms
15,776 KB
testcase_34 AC 115 ms
15,448 KB
testcase_35 AC 118 ms
18,552 KB
testcase_36 AC 141 ms
18,560 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 1000000
//#define MOD 1000000007
#define MOD 998244353
//#define INF 1000000000
#define INF 1000000000000000000

class BIT{
  int n;
  vector<ll> a;
public:
  BIT(int n_):n(n_){
    a.resize(n+1,0);
  }
  void add(int i,ll x){
    while(i<=n){
      a[i]+=x;
      i+=i&(-i);
    }
  }
  ll sum(int i){
    ll ret=0;
    while(i>0){
      ret+=a[i];
      i-=i&(-i);
    }
    return ret;
  }
  ll sum(int l,int r){
    l--;
    return sum(r)-sum(l);
  }
};

int main(){
  int N;
  cin>>N;
  vector<ll> A(N);
  for(int i=0;i<N;i++){
    cin>>A[i];
  }
  vector<ll> nums=A;
  sort(nums.begin(),nums.end());
  nums.erase(unique(nums.begin(),nums.end()),nums.end());
  int n=nums.size();

  ll ans=0;

  BIT tree1(n);
  vector<ll> cnt1(N,0);
  for(int i=N-1;i>=0;i--){
    int k=lower_bound(nums.begin(),nums.end(),A[i])-nums.begin();
    tree1.add(k+1,1);
    cnt1[i]=tree1.sum(k);
  }
  vector<pair<ll,int>> a(N);
  for(int i=0;i<N;i++){
    a[i].first=A[i];
    a[i].second=i;
  }
  sort(a.begin(),a.end());
  BIT sum1(N);
  for(int j=0;j<N;j++){
    int i=a[j].second;
    ll x=sum1.sum(i+1,N);
    x%=MOD;
    ans+=A[i]*x%MOD;
    ans%=MOD;
    sum1.add(i+1,cnt1[i]);
  }
  //cout<<ans<<endl;

  BIT tree2(n);
  vector<ll> cnt2(N,0);
  for(int i=0;i<N;i++){
    int k=lower_bound(nums.begin(),nums.end(),A[i])-nums.begin();
    tree2.add(k+1,1);
    cnt2[i]=tree2.sum(k+2,n);
  }
  BIT sum2(N);
  for(int j=N-1;j>=0;j--){
    int i=a[j].second;
    ll x=sum2.sum(i);
    x%=MOD;
    ans+=A[i]*x%MOD;
    ans%=MOD;
    sum2.add(i+1,cnt2[i]);
  }
  //cout<<ans<<endl;
  for(int i=0;i<N;i++){
    //cout<<cnt1[i]<<" "<<cnt2[i]<<endl;
    ans+=(cnt1[i]*cnt2[i]%MOD)*A[i]%MOD;
    ans%=MOD;
  }
  cout<<ans<<endl;
}
0