結果

問題 No.1300 Sum of Inversions
ユーザー chocono2230chocono2230
提出日時 2020-11-28 10:32:12
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 324 ms / 2,000 ms
コード長 1,590 bytes
コンパイル時間 4,976 ms
コンパイル使用メモリ 274,756 KB
実行使用メモリ 20,608 KB
最終ジャッジ日時 2024-09-12 22:13:45
合計ジャッジ時間 13,181 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 237 ms
16,640 KB
testcase_04 AC 232 ms
16,384 KB
testcase_05 AC 174 ms
14,080 KB
testcase_06 AC 270 ms
18,176 KB
testcase_07 AC 263 ms
17,536 KB
testcase_08 AC 299 ms
18,944 KB
testcase_09 AC 292 ms
18,816 KB
testcase_10 AC 140 ms
12,288 KB
testcase_11 AC 143 ms
12,416 KB
testcase_12 AC 234 ms
16,384 KB
testcase_13 AC 227 ms
16,000 KB
testcase_14 AC 324 ms
20,352 KB
testcase_15 AC 304 ms
18,816 KB
testcase_16 AC 251 ms
16,896 KB
testcase_17 AC 137 ms
11,904 KB
testcase_18 AC 165 ms
13,184 KB
testcase_19 AC 200 ms
14,848 KB
testcase_20 AC 203 ms
14,976 KB
testcase_21 AC 201 ms
15,104 KB
testcase_22 AC 178 ms
14,080 KB
testcase_23 AC 283 ms
18,176 KB
testcase_24 AC 190 ms
14,336 KB
testcase_25 AC 154 ms
12,928 KB
testcase_26 AC 151 ms
12,672 KB
testcase_27 AC 175 ms
13,824 KB
testcase_28 AC 306 ms
19,328 KB
testcase_29 AC 193 ms
14,976 KB
testcase_30 AC 295 ms
18,816 KB
testcase_31 AC 180 ms
14,208 KB
testcase_32 AC 189 ms
14,464 KB
testcase_33 AC 43 ms
8,064 KB
testcase_34 AC 99 ms
8,064 KB
testcase_35 AC 184 ms
20,608 KB
testcase_36 AC 204 ms
20,608 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(ri,n) for(int ri = (int)(n-1); ri >= 0; ri--)
#define rep2(i,x,n) for(int i = (int)(x); i < (int)(n); i++)
#define rrep2(ri,x,n) for(int ri = (int)(n-1); ri >= (int)(x); ri--)
#define repit(itr,x) for(auto itr = x.begin(); itr != x.end(); itr++)
#define rrepit(ritr,x) for(auto ritr = x.rbegin(); ritr != x.rend(); ritr++)
#define ALL(x) x.begin(), x.end()
using ll = long long;
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;

int main(){
  int n;
  cin >> n;
  vector<int> a(n);
  rep(i, n) cin >> a.at(i);
  map<int, int> mp;
  rep(i, n){
    mp[a.at(i)] = -1;
  }
  int c = 0;
  repit(itr, mp){
    itr->second = c;
    c++;
  }
  vector<int> b(n);
  rep(i, n){
    b.at(i) = mp[a.at(i)];
  }
  int m = mp.size();
  fenwick_tree<int> fwi(m), fwi2(m);
  fenwick_tree<mint> fwm(m), fwm2(m);
  vector<int> vil(n, 0), vir(n, 0);
  vector<mint> vml(n, 0), vmr(n, 0);
  rep(i, n){
    vil.at(i) = fwi.sum(b.at(i)+1, m);
    vml.at(i) = fwm.sum(b.at(i)+1, m);
    fwi.add(b.at(i), 1);
    fwm.add(b.at(i), a.at(i));
  }
  rrep(i, n){
    vir.at(i) = fwi2.sum(0, b.at(i));
    vmr.at(i) = fwm2.sum(0, b.at(i));
    fwi2.add(b.at(i), 1);
    fwm2.add(b.at(i), a.at(i));
  }
  mint ans = 0;
  rep(i, n){
    if(vil.at(i) != 0 && vir.at(i) != 0){
      mint add = (mint)vil.at(i) * vmr.at(i);
      add += (mint)vir.at(i) * vml.at(i);
      add += (mint)vir.at(i) * vil.at(i) * a.at(i);
      ans += add;
    }
  }
  cout << ans.val() << endl;
  return 0;
}
0