結果
問題 | No.1300 Sum of Inversions |
ユーザー | sigma425 |
提出日時 | 2020-11-28 03:12:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 153 ms / 2,000 ms |
コード長 | 5,073 bytes |
コンパイル時間 | 2,491 ms |
コンパイル使用メモリ | 215,336 KB |
実行使用メモリ | 15,748 KB |
最終ジャッジ日時 | 2024-09-12 21:53:27 |
合計ジャッジ時間 | 7,521 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,940 KB |
testcase_03 | AC | 112 ms
12,800 KB |
testcase_04 | AC | 109 ms
12,572 KB |
testcase_05 | AC | 88 ms
10,944 KB |
testcase_06 | AC | 127 ms
13,952 KB |
testcase_07 | AC | 121 ms
13,480 KB |
testcase_08 | AC | 134 ms
14,524 KB |
testcase_09 | AC | 135 ms
14,608 KB |
testcase_10 | AC | 71 ms
9,600 KB |
testcase_11 | AC | 73 ms
9,856 KB |
testcase_12 | AC | 110 ms
12,420 KB |
testcase_13 | AC | 110 ms
12,416 KB |
testcase_14 | AC | 153 ms
15,488 KB |
testcase_15 | AC | 149 ms
14,464 KB |
testcase_16 | AC | 119 ms
13,184 KB |
testcase_17 | AC | 70 ms
9,404 KB |
testcase_18 | AC | 84 ms
10,368 KB |
testcase_19 | AC | 98 ms
11,648 KB |
testcase_20 | AC | 99 ms
11,776 KB |
testcase_21 | AC | 97 ms
11,676 KB |
testcase_22 | AC | 88 ms
11,044 KB |
testcase_23 | AC | 125 ms
13,952 KB |
testcase_24 | AC | 91 ms
11,136 KB |
testcase_25 | AC | 76 ms
10,164 KB |
testcase_26 | AC | 76 ms
9,984 KB |
testcase_27 | AC | 85 ms
10,644 KB |
testcase_28 | AC | 139 ms
14,720 KB |
testcase_29 | AC | 97 ms
11,572 KB |
testcase_30 | AC | 136 ms
14,464 KB |
testcase_31 | AC | 87 ms
11,136 KB |
testcase_32 | AC | 93 ms
11,392 KB |
testcase_33 | AC | 27 ms
12,672 KB |
testcase_34 | AC | 40 ms
12,672 KB |
testcase_35 | AC | 80 ms
15,616 KB |
testcase_36 | AC | 85 ms
15,748 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; using uint = unsigned int; #define rep(i,n) for(int i=0;i<int(n);i++) #define rep1(i,n) for(int i=1;i<=int(n);i++) #define per(i,n) for(int i=int(n)-1;i>=0;i--) #define per1(i,n) for(int i=int(n);i>0;i--) #define all(c) c.begin(),c.end() #define si(x) int(x.size()) #define pb emplace_back #define fs first #define sc second template<class T> using V = vector<T>; template<class T> using VV = vector<vector<T>>; template<class T,class U> void chmax(T& x, U y){if(x<y) x=y;} template<class T,class U> void chmin(T& x, U y){if(y<x) x=y;} template<class T> void mkuni(V<T>& v){sort(all(v));v.erase(unique(all(v)),v.end());} template<class S,class T> ostream& operator<<(ostream& o,const pair<S,T> &p){ return o<<"("<<p.fs<<","<<p.sc<<")"; } template<class T> ostream& operator<<(ostream& o,const vector<T> &vc){ o<<"{"; for(const T& v:vc) o<<v<<","; o<<"}"; return o; } constexpr ll TEN(int n) { return (n == 0) ? 1 : 10 * TEN(n-1); } #ifdef LOCAL #define show(x) cerr << "LINE" << __LINE__ << " : " << #x << " = " << (x) << endl void dmpr(ostream& os){os<<endl;} template<class T,class... Args> void dmpr(ostream&os,const T&t,const Args&... args){ os<<t<<" ~ "; dmpr(os,args...); } #define shows(...) cerr << "LINE" << __LINE__ << " : ";dmpr(cerr,##__VA_ARGS__) #define dump(x) cerr << "LINE" << __LINE__ << " : " << #x << " = {"; \ for(auto v: x) cerr << v << ","; cerr << "}" << endl; #else #define show(x) void(0) #define dump(x) void(0) #define shows(...) void(0) #endif struct BIT{ using D = ll; int N; vector<D> bit; BIT(int n){ //nは区間幅!!! N=n; bit.assign(n+1,0); } BIT(){} void init(int n){ N=n; bit.assign(n+1,0); } D sum(int i){ //i個の和a[0]+..a[i-1] D s=0; while(i>0){ s+=bit[i]; i-=(i&-i); } return s; } D val(int i){ //return a[i] (0-ind) return sum(i+1)-sum(i); } void add(int i,D x){ //a[i]+=x (0-indexed) i++; while(i<=N){ bit[i]+=x; i+=(i&-i); } } }; template<unsigned int mod_> struct ModInt{ using uint = unsigned int; using ll = long long; using ull = unsigned long long; constexpr static uint mod = mod_; uint v; ModInt():v(0){} ModInt(ll _v):v(normS(_v%mod+mod)){} explicit operator bool() const {return v!=0;} static uint normS(const uint &x){return (x<mod)?x:x-mod;} // [0 , 2*mod-1] -> [0 , mod-1] static ModInt make(const uint &x){ModInt m; m.v=x; return m;} ModInt operator+(const ModInt& b) const { return make(normS(v+b.v));} ModInt operator-(const ModInt& b) const { return make(normS(v+mod-b.v));} ModInt operator-() const { return make(normS(mod-v)); } ModInt operator*(const ModInt& b) const { return make((ull)v*b.v%mod);} ModInt operator/(const ModInt& b) const { return *this*b.inv();} ModInt& operator+=(const ModInt& b){ return *this=*this+b;} ModInt& operator-=(const ModInt& b){ return *this=*this-b;} ModInt& operator*=(const ModInt& b){ return *this=*this*b;} ModInt& operator/=(const ModInt& b){ return *this=*this/b;} ModInt& operator++(int){ return *this=*this+1;} ModInt& operator--(int){ return *this=*this-1;} ll extgcd(ll a,ll b,ll &x,ll &y) const{ ll p[]={a,1,0},q[]={b,0,1}; while(*q){ ll t=*p/ *q; rep(i,3) swap(p[i]-=t*q[i],q[i]); } if(p[0]<0) rep(i,3) p[i]=-p[i]; x=p[1],y=p[2]; return p[0]; } ModInt inv() const { ll x,y; extgcd(v,mod,x,y); return make(normS(x+mod)); } ModInt pow(ll p) const { if(p<0) return inv().pow(-p); ModInt a = 1; ModInt x = *this; while(p){ if(p&1) a *= x; x *= x; p >>= 1; } return a; } bool operator==(const ModInt& b) const { return v==b.v;} bool operator!=(const ModInt& b) const { return v!=b.v;} friend istream& operator>>(istream &o,ModInt& x){ ll tmp; o>>tmp; x=ModInt(tmp); return o; } friend ostream& operator<<(ostream &o,const ModInt& x){ return o<<x.v;} }; using mint = ModInt<998244353>; //using mint = ModInt<1000000007>; V<mint> fact,ifact,invs; mint Choose(int a,int b){ if(b<0 || a<b) return 0; return fact[a] * ifact[b] * ifact[a-b]; } void InitFact(int N){ //[0,N] N++; fact.resize(N); ifact.resize(N); invs.resize(N); fact[0] = 1; rep1(i,N-1) fact[i] = fact[i-1] * i; ifact[N-1] = fact[N-1].inv(); for(int i=N-2;i>=0;i--) ifact[i] = ifact[i+1] * (i+1); rep1(i,N-1) invs[i] = fact[i-1] * ifact[i]; } int main(){ cin.tie(0); ios::sync_with_stdio(false); //DON'T USE scanf/printf/puts !! cout << fixed << setprecision(20); int N; cin >> N; V<ll> A(N); rep(i,N) cin >> A[i]; reverse(all(A)); V<ll> as = A; mkuni(as); V<ll> L(N),LL(N),R(N),RR(N); { BIT n(si(as)),s(si(as)); rep(i,N){ int x = lower_bound(all(as),A[i]) - as.begin(); L[i] = n.sum(x); LL[i] = s.sum(x); n.add(x,1); s.add(x,L[i]); } } { BIT n(si(as)),s(si(as)); per(i,N){ int x = lower_bound(all(as),A[i]) - as.begin(); R[i] = n.sum(si(as)) - n.sum(x+1); RR[i] = s.sum(si(as)) - s.sum(x+1); n.add(x,1); s.add(x,R[i]); } } mint ans = 0; rep(i,N){ ans += mint(A[i]) * (LL[i] + L[i]*R[i] + RR[i]); } cout << ans << endl; }