結果

問題 No.1300 Sum of Inversions
ユーザー ocvret_ocvret_
提出日時 2020-11-27 22:57:33
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,151 bytes
コンパイル時間 2,245 ms
コンパイル使用メモリ 187,792 KB
実行使用メモリ 42,056 KB
最終ジャッジ日時 2024-07-26 19:15:54
合計ジャッジ時間 18,613 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 520 ms
38,748 KB
testcase_04 AC 490 ms
38,464 KB
testcase_05 AC 396 ms
24,384 KB
testcase_06 AC 558 ms
40,120 KB
testcase_07 AC 543 ms
39,396 KB
testcase_08 AC 603 ms
40,672 KB
testcase_09 AC 604 ms
40,576 KB
testcase_10 AC 312 ms
22,744 KB
testcase_11 AC 316 ms
22,892 KB
testcase_12 AC 487 ms
38,492 KB
testcase_13 AC 487 ms
38,248 KB
testcase_14 AC 670 ms
41,592 KB
testcase_15 AC 605 ms
40,536 KB
testcase_16 AC 518 ms
38,888 KB
testcase_17 AC 302 ms
22,520 KB
testcase_18 AC 356 ms
23,708 KB
testcase_19 AC 433 ms
37,248 KB
testcase_20 AC 452 ms
37,504 KB
testcase_21 AC 439 ms
37,420 KB
testcase_22 AC 388 ms
24,368 KB
testcase_23 AC 578 ms
39,992 KB
testcase_24 AC 406 ms
24,664 KB
testcase_25 AC 333 ms
23,460 KB
testcase_26 AC 335 ms
23,240 KB
testcase_27 AC 369 ms
24,056 KB
testcase_28 AC 612 ms
40,864 KB
testcase_29 AC 431 ms
37,476 KB
testcase_30 AC 607 ms
40,648 KB
testcase_31 AC 379 ms
24,356 KB
testcase_32 AC 408 ms
24,616 KB
testcase_33 AC 45 ms
6,944 KB
testcase_34 AC 101 ms
6,948 KB
testcase_35 AC 362 ms
41,984 KB
testcase_36 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
// #include<atcoder/all>

using namespace std;
// using namespace atcoder;
using ll = long long;
using ull = unsigned long long;
using P = pair<int,int>;
#define rep(i,n) for(ll i = 0;i < (ll)n;i++)
#define ALL(x) (x).begin(),(x).end()
#define MOD 1000000007

template<typename T>
class Segtree{
  public:
  using F = function<T(T,T)>;
  int n;
  F f;
  T ti;
  vector<T> dat;
  Segtree(){};
  Segtree(F f,T ti): f(f),ti(ti){}
  void init(int N){
    n = 1;
    while(n < N)n <<= 1;
    dat.assign(n << 1,ti);
  }
  void build(const vector<T> &v){
    int N = v.size();
    init(N);
    rep(i,N)dat[n+i] = v[i];
    for(int i = n-1;i >= 0;i--)dat[i] = f(dat[(i << 1)|0],dat[(i << 1)|1]);
  }
  void set(int k,T x){
    dat[k+=n] = x;
    while(k >>= 1){
      dat[k] = f(dat[(k << 1)|0],dat[(k << 1)|1]);
    }
  }
  T get(int l,int r){
    T vl = ti,vr = ti;
    for(l += n,r += n;l < r;l >>= 1,r >>= 1){
      if(l & 1)vl = f(vl,dat[l++]);
      if(r & 1)vr = f(dat[--r],vr);
    }
    return f(vl,vr);
  }
};
const int mod = 998244353;

int main(){
  
  int n;
  cin >> n;
  vector<int> v(n);
  vector<int> w(n);
  rep(i,n)cin >> v[i],w[i] = v[i];
  sort(ALL(w));
  w.erase(unique(ALL(w)),w.end());
  map<int,int> mp;
  int m = w.size();
  rep(i,m)mp[w[i]] = i;
  auto f = [&](pair<ll,ll> a,pair<ll,ll> b){return pair<ll,ll>{(a.first+b.first)%mod,a.second+b.second};};
  Segtree<pair<ll,ll>> one(f,{0,0}),two(f,{0,0}),three(f,{0,0});
  one.build(vector<pair<ll,ll>>(m,{0,0}));
  two.build(vector<pair<ll,ll>>(m,{0,0}));
  three.build(vector<pair<ll,ll>>(m,{0,0}));
  rep(i,n){
    int k = mp[v[i]];
    {
      pair<ll,ll> x = one.get(k,k+1);
      x.first += v[i];x.second++;
      x.first %= mod;
      one.set(k,x);
    }
    {
      pair<ll,ll> x = two.get(k,k+1);
      pair<ll,ll> y = one.get(k+1,m);
      two.set(k,{(x.first+y.first+y.second*v[i]%mod)%mod,x.second+y.second});
    }
    {
      pair<ll,ll> x = three.get(k,k+1);
      pair<ll,ll> y = two.get(k+1,m);
      three.set(k,{(x.first+y.first+y.second*v[i]%mod)%mod,x.second+y.second});
    }
  }
  cout << three.get(0,m).first << "\n";
  



  return 0;
}
0