結果

問題 No.1300 Sum of Inversions
ユーザー chocono2230chocono2230
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 137 ms
12,288 KB
testcase_04 AC 131 ms
12,032 KB
testcase_05 AC 112 ms
10,496 KB
testcase_06 AC 153 ms
13,312 KB
testcase_07 AC 153 ms
13,056 KB
testcase_08 AC 165 ms
13,952 KB
testcase_09 AC 161 ms
13,824 KB
testcase_10 AC 87 ms
9,344 KB
testcase_11 AC 89 ms
9,344 KB
testcase_12 AC 133 ms
12,032 KB
testcase_13 AC 135 ms
11,904 KB
testcase_14 AC 177 ms
14,848 KB
testcase_15 AC 163 ms
13,824 KB
testcase_16 AC 139 ms
12,544 KB
testcase_17 AC 87 ms
9,088 KB
testcase_18 AC 98 ms
10,112 KB
testcase_19 AC 117 ms
11,264 KB
testcase_20 AC 119 ms
11,264 KB
testcase_21 AC 115 ms
11,264 KB
testcase_22 AC 109 ms
10,496 KB
testcase_23 AC 154 ms
13,440 KB
testcase_24 AC 112 ms
10,752 KB
testcase_25 AC 93 ms
9,728 KB
testcase_26 AC 92 ms
9,600 KB
testcase_27 AC 107 ms
10,368 KB
testcase_28 AC 170 ms
14,080 KB
testcase_29 AC 119 ms
11,136 KB
testcase_30 AC 163 ms
13,824 KB
testcase_31 AC 110 ms
10,624 KB
testcase_32 AC 114 ms
10,880 KB
testcase_33 AC 88 ms
14,976 KB
testcase_34 AC 155 ms
15,104 KB
testcase_35 AC 123 ms
14,976 KB
testcase_36 AC 151 ms
14,976 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