結果
問題 | No.1300 Sum of Inversions |
ユーザー | umezo |
提出日時 | 2020-11-28 04:31:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 332 ms / 2,000 ms |
コード長 | 1,583 bytes |
コンパイル時間 | 3,274 ms |
コンパイル使用メモリ | 203,424 KB |
最終ジャッジ日時 | 2025-01-16 09:08:03 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#define rep(i, n) for (int i = 0; i < (int)(n); i++) #define ALL(v) v.begin(), v.end() typedef long long ll; #include <bits/stdc++.h> using namespace std; const ll MOD=998244353; template <typename T> vector<T> compress(vector<T> &X){ vector<T> vals=X; sort(vals.begin(),vals.end()); vals.erase(unique(vals.begin(), vals.end()),vals.end()); for(int i=0;i<(int)X.size(); i++){ X[i]=lower_bound(vals.begin(),vals.end(),X[i])-vals.begin()+1; } return vals; } template<typename T> struct BIT{ private: vector<T> A; const int n; public: BIT(int _n) : A(_n+1,0), n(_n){} T sum(int i){ T s=0; while(i>0){ s=(s+A[i])%MOD; i-=i&-i; } return s; } T sum(int i,int j){ return (sum(j)-sum(i-1)+MOD)%MOD; } void add(int i,T x){ while(i<=n){ A[i]=(A[i]+x)%MOD; i+=i&-i; } } }; int main(){ int n; cin>>n; vector<ll> A(n),B(n); rep(i,n) cin>>A[i]; B=A; compress(B); BIT<ll> bitlcnt(n),bitlsum(n),bitrcnt(n),bitrsum(n); vector<ll> Lcnt(n),Rcnt(n),Lsum(n),Rsum(n); ll tmp=0; rep(i,n){ tmp+=A[i]; Lcnt[i]=bitlcnt.sum(B[i]+1,n); bitlcnt.add(B[i],1); Lsum[i]=bitlsum.sum(B[i]+1,n); bitlsum.add(B[i],A[i]); } for(int i=n-1;i>=0;i--){ Rcnt[i]=bitrcnt.sum(B[i]-1); bitrcnt.add(B[i],1); Rsum[i]=bitrsum.sum(B[i]-1); bitrsum.add(B[i],A[i]); } ll ans=0; rep(i,n){ ans=(ans+Rcnt[i]*Lcnt[i]%MOD*A[i]%MOD)%MOD; ans=(ans+Rsum[i]*Lcnt[i]%MOD)%MOD; ans=(ans+Lsum[i]*Rcnt[i]%MOD)%MOD; } cout<<ans<<endl; return 0; }