結果

問題 No.1300 Sum of Inversions
ユーザー tko919tko919
提出日時 2021-08-27 20:58:00
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 197 ms / 2,000 ms
コード長 3,312 bytes
コンパイル時間 2,340 ms
コンパイル使用メモリ 212,096 KB
実行使用メモリ 8,152 KB
最終ジャッジ日時 2024-11-20 23:52:39
合計ジャッジ時間 9,869 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
6,820 KB
testcase_01 AC 4 ms
6,820 KB
testcase_02 AC 4 ms
6,820 KB
testcase_03 AC 151 ms
7,720 KB
testcase_04 AC 153 ms
7,700 KB
testcase_05 AC 122 ms
7,348 KB
testcase_06 AC 177 ms
7,792 KB
testcase_07 AC 166 ms
7,760 KB
testcase_08 AC 186 ms
7,956 KB
testcase_09 AC 179 ms
7,948 KB
testcase_10 AC 101 ms
7,268 KB
testcase_11 AC 102 ms
7,268 KB
testcase_12 AC 148 ms
7,712 KB
testcase_13 AC 145 ms
7,568 KB
testcase_14 AC 197 ms
8,020 KB
testcase_15 AC 184 ms
7,952 KB
testcase_16 AC 158 ms
7,608 KB
testcase_17 AC 97 ms
7,252 KB
testcase_18 AC 111 ms
7,436 KB
testcase_19 AC 132 ms
7,388 KB
testcase_20 AC 135 ms
7,512 KB
testcase_21 AC 133 ms
7,524 KB
testcase_22 AC 121 ms
7,480 KB
testcase_23 AC 176 ms
7,776 KB
testcase_24 AC 124 ms
7,496 KB
testcase_25 AC 107 ms
7,552 KB
testcase_26 AC 105 ms
7,284 KB
testcase_27 AC 118 ms
7,464 KB
testcase_28 AC 185 ms
8,100 KB
testcase_29 AC 133 ms
7,516 KB
testcase_30 AC 178 ms
7,956 KB
testcase_31 AC 123 ms
7,480 KB
testcase_32 AC 125 ms
7,496 KB
testcase_33 AC 102 ms
8,028 KB
testcase_34 AC 158 ms
8,028 KB
testcase_35 AC 122 ms
8,152 KB
testcase_36 AC 143 ms
8,028 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;

//template
#define rep(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define ALL(v) (v).begin(),(v).end()
using ll=long long int;
const int inf = 0x3fffffff; const ll INF = 0x1fffffffffffffff; const double eps=1e-12;
template<typename T>inline bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
template<typename T>inline bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}

template<int mod=998244353>struct fp {
   int v; static int get_mod(){return mod;}
   int inv() const{
      int tmp,a=v,b=mod,x=1,y=0;
      while(b)tmp=a/b,a-=tmp*b,swap(a,b),x-=tmp*y,swap(x,y);
      if(x<0){x+=mod;} return x;
   }
   fp(ll x=0){init(x%mod+mod);}
   fp& init(int x){v=(x<mod?x:x-mod); return *this;}
   fp operator-()const{return fp()-*this;}
   fp pow(ll t){fp res=1,b=*this; while(t){if(t&1)res*=b;b*=b;t>>=1;} return res;}
   fp& operator+=(const fp& x){return init(v+x.v);}
   fp& operator-=(const fp& x){return init(v+mod-x.v);}
   fp& operator*=(const fp& x){v=ll(v)*x.v%mod; return *this;}
   fp& operator/=(const fp& x){v=ll(v)*x.inv()%mod; return *this;}
   fp operator+(const fp& x)const{return fp(*this)+=x;}
   fp operator-(const fp& x)const{return fp(*this)-=x;}
   fp operator*(const fp& x)const{return fp(*this)*=x;}
   fp operator/(const fp& x)const{return fp(*this)/=x;}
   bool operator==(const fp& x)const{return v==x.v;}
   bool operator!=(const fp& x)const{return v!=x.v;}
}; using Fp=fp<>;
template<typename T>struct factorial {
   vector<T> Fact,Finv,Inv;
   factorial(int maxx){
      Fact.resize(maxx); Finv.resize(maxx); Inv.resize(maxx);
      Fact[0]=Fact[1]=Finv[0]=Finv[1]=Inv[1]=1;
      rep(i,2,maxx){Fact[i]=Fact[i-1]*i;} Finv[maxx-1]=Fact[maxx-1].inv();
      for(int i=maxx-1;i>=2;i--){Finv[i-1]=Finv[i]*i; Inv[i]=Finv[i]*Fact[i-1];}
   }
   T fact(int n,bool inv=0){if(n<0)return 0; return (inv?Finv[n]:Fact[n]);}
   T inv(int n){if(n<0)return 0; return Inv[n];}
   T nPr(int n,int r,bool inv=0){if(n<0||n<r||r<0)return 0; return fact(n,inv)*fact(n-r,inv^1);}
   T nCr(int n,int r,bool inv=0){if(n<0||n<r||r<0)return 0; return fact(n,inv)*fact(r,inv^1)*fact(n-r,inv^1);}
   T nHr(int n,int r,bool inv=0){return nCr(n+r-1,r,inv);}
};

template<typename T>struct BIT{
   int n; T all=0; vector<T> val;
   BIT(int _n):n(_n),val(_n+10){}
   void clear(){val.assign(n+10,0); all=T();}
   void add(int i,T x){
      for(i++;i<=n;i+=(i&-i))val[i]+=x;
      all+=x;
   }
   T sum(int i){
      T res=0;
      for(i++;i;i-=(i&-i))res+=val[i];
      return res;
   }
};

int main(){
   int n;
   cin>>n;
   vector<int> a(n),vs;
   rep(i,0,n){
      cin>>a[i];
      vs.push_back(a[i]);
   }
   sort(ALL(vs));
   vs.erase(unique(ALL(vs)),vs.end());
   rep(i,0,n)a[i]=lower_bound(ALL(vs),a[i])-vs.begin();

   BIT<Fp> L(201010),R(201010),Lcnt(201010),Rcnt(201010);
   rep(i,0,n){
      R.add(a[i],vs[a[i]]);
      Rcnt.add(a[i],1);
   }
   Fp res;
   rep(i,0,n-1){
      R.add(a[i],-vs[a[i]]);
      Rcnt.add(a[i],-1);
      if(i){
         L.add(a[i-1],vs[a[i-1]]);
         Lcnt.add(a[i-1],1);
         res+=(L.all-L.sum(a[i]))*Rcnt.sum(a[i]-1);
         res+=R.sum(a[i]-1)*(Lcnt.all-Lcnt.sum(a[i]));
         res+=(Lcnt.all-Lcnt.sum(a[i]))*Rcnt.sum(a[i]-1)*vs[a[i]];
      }
   }
   cout<<res.v<<'\n';
   return 0;
}
0