結果

問題 No.1300 Sum of Inversions
コンテスト
ユーザー chocono2230
提出日時 2020-11-28 00:26:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 177 ms / 2,000 ms
コード長 3,102 bytes
コンパイル時間 4,926 ms
コンパイル使用メモリ 262,164 KB
最終ジャッジ日時 2025-01-16 08:51:48
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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