結果
問題 | No.1300 Sum of Inversions |
ユーザー | chocorusk |
提出日時 | 2020-11-27 22:37:38 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 342 ms / 2,000 ms |
コード長 | 2,571 bytes |
コンパイル時間 | 1,573 ms |
コンパイル使用メモリ | 134,512 KB |
最終ジャッジ日時 | 2025-01-16 08:00:30 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#include <cstdio>#include <cstring>#include <iostream>#include <string>#include <cmath>#include <bitset>#include <vector>#include <map>#include <set>#include <queue>#include <deque>#include <algorithm>#include <complex>#include <unordered_map>#include <unordered_set>#include <random>#include <cassert>#include <fstream>#include <utility>#include <functional>#include <time.h>#include <stack>#include <array>#include <list>#define popcount __builtin_popcountusing namespace std;typedef long long ll;typedef pair<int, int> P;const ll MOD=998244353;ll powmod(ll a, ll k){ll ap=a, ans=1;while(k){if(k&1){ans*=ap;ans%=MOD;}ap=ap*ap;ap%=MOD;k>>=1;}return ans;}ll inv(ll a){return powmod(a, MOD-2);}ll f[2000010], invf[2000010];void fac(int n){f[0]=1;for(ll i=1; i<=n; i++) f[i]=f[i-1]*i%MOD;invf[n]=inv(f[n]);for(ll i=n-1; i>=0; i--) invf[i]=invf[i+1]*(i+1)%MOD;}ll comb(int x, int y){if(!(0<=y && y<=x)) return 0;return f[x]*invf[y]%MOD*invf[x-y]%MOD;}template<typename T>struct BIT{vector<T> bit;int size;BIT(int n):size(n), bit(n+1, 0){}T sum(int i){ //[0, i)T s=0;while(i>0){s+=bit[i];s%=MOD;i-=(i&(-i));}return s;}T sum(int l, int r){ //[l, r)return (sum(r)-sum(l)+MOD)%MOD;}void add(int i, T x){i++;while(i<=size){bit[i]+=x;bit[i]%=MOD;i+=(i&(-i));}}};int main(){int n; cin>>n;ll a[200020];vector<ll> v(n);for(int i=0; i<n; i++){cin>>a[i];//a[i]%=MOD;v[i]=a[i];}sort(v.begin(), v.end());v.erase(unique(v.begin(), v.end()), v.end());int m=v.size();for(int i=0; i<n; i++){a[i]=lower_bound(v.begin(), v.end(), a[i])-v.begin();}BIT<ll> bit(m), bit2(m), bit3(m), bit4(m), bit5(m);ll ans=0;for(int i=0; i<n; i++) bit2.add(a[i], 1);for(int i=0; i<n; i++){bit2.add(a[i], MOD-1);ll s=bit.sum(a[i]+1, m);(ans+=v[a[i]]%MOD*s%MOD*bit2.sum(a[i]))%=MOD;(ans+=v[a[i]]%MOD*bit3.sum(a[i]+1, m))%=MOD;bit.add(a[i], 1);bit3.add(a[i], s);}reverse(a, a+n);for(int i=0; i<n; i++){ll s=bit4.sum(a[i]);(ans+=v[a[i]]%MOD*bit5.sum(a[i]))%=MOD;bit4.add(a[i], 1);bit5.add(a[i], s);}cout<<ans<<endl;return 0;}