結果

問題 No.1300 Sum of Inversions
ユーザー monnumonnu
提出日時 2021-07-11 15:44:50
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 234 ms / 2,000 ms
コード長 1,876 bytes
コンパイル時間 2,162 ms
コンパイル使用メモリ 182,536 KB
実行使用メモリ 18,816 KB
最終ジャッジ日時 2024-07-02 03:05:46
合計ジャッジ時間 10,197 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 176 ms
15,232 KB
testcase_04 AC 177 ms
14,976 KB
testcase_05 AC 142 ms
12,928 KB
testcase_06 AC 199 ms
16,768 KB
testcase_07 AC 191 ms
16,128 KB
testcase_08 AC 214 ms
17,408 KB
testcase_09 AC 210 ms
17,408 KB
testcase_10 AC 116 ms
11,392 KB
testcase_11 AC 116 ms
11,392 KB
testcase_12 AC 174 ms
14,976 KB
testcase_13 AC 173 ms
14,720 KB
testcase_14 AC 234 ms
18,560 KB
testcase_15 AC 214 ms
17,280 KB
testcase_16 AC 184 ms
15,744 KB
testcase_17 AC 111 ms
11,008 KB
testcase_18 AC 131 ms
12,160 KB
testcase_19 AC 153 ms
13,696 KB
testcase_20 AC 157 ms
13,952 KB
testcase_21 AC 155 ms
13,952 KB
testcase_22 AC 141 ms
12,928 KB
testcase_23 AC 199 ms
16,640 KB
testcase_24 AC 145 ms
13,312 KB
testcase_25 AC 124 ms
11,904 KB
testcase_26 AC 122 ms
11,776 KB
testcase_27 AC 137 ms
12,800 KB
testcase_28 AC 218 ms
17,792 KB
testcase_29 AC 156 ms
13,824 KB
testcase_30 AC 215 ms
17,280 KB
testcase_31 AC 142 ms
13,056 KB
testcase_32 AC 149 ms
13,312 KB
testcase_33 AC 68 ms
15,744 KB
testcase_34 AC 123 ms
15,744 KB
testcase_35 AC 131 ms
18,816 KB
testcase_36 AC 152 ms
18,816 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