結果

問題 No.1300 Sum of Inversions
ユーザー chocono2230chocono2230
提出日時 2020-11-28 00:26:31
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 149 ms / 2,000 ms
コード長 3,102 bytes
コンパイル時間 5,016 ms
コンパイル使用メモリ 270,712 KB
実行使用メモリ 15,224 KB
最終ジャッジ日時 2023-10-09 22:55:40
合計ジャッジ時間 9,746 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,352 KB
testcase_01 AC 2 ms
4,356 KB
testcase_02 AC 1 ms
4,352 KB
testcase_03 AC 116 ms
12,028 KB
testcase_04 AC 108 ms
11,752 KB
testcase_05 AC 89 ms
10,424 KB
testcase_06 AC 125 ms
13,200 KB
testcase_07 AC 120 ms
12,888 KB
testcase_08 AC 136 ms
13,612 KB
testcase_09 AC 133 ms
13,608 KB
testcase_10 AC 73 ms
9,068 KB
testcase_11 AC 75 ms
9,120 KB
testcase_12 AC 110 ms
11,980 KB
testcase_13 AC 110 ms
11,812 KB
testcase_14 AC 149 ms
14,624 KB
testcase_15 AC 136 ms
13,556 KB
testcase_16 AC 116 ms
12,356 KB
testcase_17 AC 72 ms
8,860 KB
testcase_18 AC 81 ms
9,924 KB
testcase_19 AC 98 ms
10,912 KB
testcase_20 AC 103 ms
10,920 KB
testcase_21 AC 100 ms
11,104 KB
testcase_22 AC 89 ms
10,388 KB
testcase_23 AC 133 ms
13,076 KB
testcase_24 AC 93 ms
10,720 KB
testcase_25 AC 77 ms
9,724 KB
testcase_26 AC 78 ms
9,340 KB
testcase_27 AC 86 ms
10,304 KB
testcase_28 AC 139 ms
13,944 KB
testcase_29 AC 97 ms
10,972 KB
testcase_30 AC 133 ms
13,564 KB
testcase_31 AC 90 ms
10,412 KB
testcase_32 AC 92 ms
10,740 KB
testcase_33 AC 75 ms
15,224 KB
testcase_34 AC 122 ms
14,652 KB
testcase_35 AC 100 ms
14,744 KB
testcase_36 AC 118 ms
14,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);
  vector<pair<int, int>> p(n);
  rep(i, n){
    p.at(i) = {a.at(i), i};
  }

  sort(ALL(p), [](const auto &l, const auto &r){
    if(l.first != r.first) return l.first > r.first;
    return l.second < r.second;
  });
  // for(auto [f, s] : p){
  //   cerr << f << ":" << s << " ";
  // }
  // cerr << endl;

  fenwick_tree<int> fwi(n), fwi2(n);
  fenwick_tree<mint> fwm(n), fwm2(n);
  vector<int> scl(n, 0), scr(n, 0);
  rep2(i, 1, n){
    if(p.at(i-1).first == p.at(i).first){
      scl.at(i) = scl.at(i-1) + 1;
    }
  }
  rrep(i, n-1){
    if(p.at(i).first == p.at(i+1).first){
      scr.at(i) = scr.at(i+1) + 1;
    }
  }
  vector<ll> vil(n, 0), vir(n, 0);
  vector<mint> vml(n, 0), vmr(n, 0);
  rep(i, n){
    auto [f, s] = p.at(i);
    vil.at(s) = fwi.sum(0, s);
    vml.at(s) = fwm.sum(0, s);
    if(scl.at(i) != 0){
      vil.at(s) -= scl.at(i);
      vml.at(s) -= (mint)f * scl.at(i);
    }
    // cerr << f << " " << s << " " << vil.at(s) << " " << vml.at(s).val() << endl;
    fwi.add(s, 1);
    fwm.add(s, f);
  }
  // cerr << endl;
  rrep(i, n){
    auto [f, s] = p.at(i);
    vir.at(s) = fwi2.sum(s, n);
    vmr.at(s) = fwm2.sum(s, n);
    if(scr.at(i) != 0){
      vir.at(s) -= scr.at(i);
      vmr.at(s) -= (mint)f * scr.at(i);
    }
    fwi2.add(s, 1);
    fwm2.add(s, f);
  }
  mint ans = 0;
  rep(i, n){
    if(vil.at(i) != 0 && vir.at(i) != 0){
      mint add = (mint)vir.at(i) * vml.at(i);
      add += (mint)vil.at(i) * vmr.at(i);
      add += (mint)vil.at(i) * vir.at(i) * a.at(i);
      ans += add;
    }
  }
  cout << ans.val() << endl;
  // rep(i, n){
  //   int co = fw.sum(0, p.at(i).second);
  //   cerr << co << ":" << p.at(i).first << ":" << p.at(i).second << " ";
  //   vm.at(p.at(i).second) = co;
  //   vl.at(p.at(i).second) = (ll) co * (co-1)/2;
  //   fw.add(p.at(i).second, 1);
  // }
  // rrep(i, n){
  //   int co = fw2.sum(p.at(i).second, n);
  //   vm.at(p.at(i).second) *= co;
  //   vr.at(p.at(i).second) = (ll) co * (co-1)/2;
  //   fw2.add(p.at(i).second, 1);
  // }
  // for(auto i : vl) cerr << i.val() << " ";
  // cerr << endl;
  // for(auto i : vm) cerr << i.val() << " ";
  // cerr << endl;
  // for(auto i : vr) cerr << i.val() << " ";
  // cerr << endl;
  // mint ans = 0;
  // rep(i, n){
  //   mint add = a.at(i);
  //   add *= (vl.at(i) + vm.at(i) + vr.at(i));
  //   cerr << (vl.at(i) + vm.at(i) + vr.at(i)).val() << " ";
  //   ans += add;
  // }
  // cout << ans.val() << endl;


  return 0;
}
0