結果

問題 No.1300 Sum of Inversions
ユーザー 👑 CleyL
提出日時 2022-10-22 01:31:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 630 ms / 2,000 ms
コード長 1,346 bytes
コンパイル時間 1,024 ms
コンパイル使用メモリ 92,420 KB
最終ジャッジ日時 2025-02-08 10:51:27
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
const long long MOD = 998244353;
//B
template<typename T>
struct BIT{//1_Indexed
int n;
vector<T> bit;
BIT(int n_):n(n_+1),bit(n,0){}
T sum(int a){
T ret = 0;
for(int i = a; i > 0; i -= i & -i) ret += bit[i];
return ret;
}
void add(int a,int w){
for(int i = a; i <= n; i += i & -i)bit[i] += w;
}
T query(int r,int l){
return sum(l-1)-sum(r-1);
}
};
//E
int main(){
int n;cin>>n;
vector<int> A(n),B(n);
for(int i = 0; n > i; i++)cin>>A[i],B[i] = A[i];
sort(B.begin(),B.end());
B.erase(unique(B.begin(),B.end()),B.end());
map<int,int> C;
for(int i = 0; B.size() > i; i++){
C[B[i]] = i+1;
}
BIT<long long> X(B.size()+1),Y(B.size()+1),Z(B.size()+1),W(B.size()+1);
for(int i = 0; n > i; i++){
Y.add(C[A[i]],1);
W.add(C[A[i]],A[i]);
}
long long ans = 0;
for(int i = 0; n > i; i++){
auto x = X.query(C[A[i]]+1,B.size()+1)%MOD;
auto y = Y.query(1,C[A[i]])%MOD;
auto z = Z.query(C[A[i]]+1,B.size()+1)%MOD;
auto w = W.query(1,C[A[i]])%MOD;
ans = (ans + ((z*y + ((A[i]*x)%MOD)*y + w*x)%MOD))%MOD;
X.add(C[A[i]],1);
Y.add(C[A[i]],-1);
Z.add(C[A[i]],A[i]);
W.add(C[A[i]],-A[i]);
}
cout << ans << endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0