結果

問題 No.1300 Sum of Inversions
ユーザー monnu
提出日時 2021-07-11 15:44:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 234 ms / 2,000 ms
コード長 1,876 bytes
コンパイル時間 2,162 ms
コンパイル使用メモリ 182,536 KB
実行使用メモリ 18,816 KB
最終ジャッジ日時 2024-07-02 03:05:46
合計ジャッジ時間 10,197 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 1000000
//#define MOD 1000000007
#define MOD 998244353
//#define INF 1000000000
#define INF 1000000000000000000
class BIT{
int n;
vector<ll> a;
public:
BIT(int n_):n(n_){
a.resize(n+1,0);
}
void add(int i,ll x){
while(i<=n){
a[i]+=x;
i+=i&(-i);
}
}
ll sum(int i){
ll ret=0;
while(i>0){
ret+=a[i];
i-=i&(-i);
}
return ret;
}
ll sum(int l,int r){
l--;
return sum(r)-sum(l);
}
};
int main(){
int N;
cin>>N;
vector<ll> A(N);
for(int i=0;i<N;i++){
cin>>A[i];
}
vector<ll> nums=A;
sort(nums.begin(),nums.end());
nums.erase(unique(nums.begin(),nums.end()),nums.end());
int n=nums.size();
ll ans=0;
BIT tree1(n);
vector<ll> cnt1(N,0);
for(int i=N-1;i>=0;i--){
int k=lower_bound(nums.begin(),nums.end(),A[i])-nums.begin();
tree1.add(k+1,1);
cnt1[i]=tree1.sum(k);
}
vector<pair<ll,int>> a(N);
for(int i=0;i<N;i++){
a[i].first=A[i];
a[i].second=i;
}
sort(a.begin(),a.end());
BIT sum1(N);
for(int j=0;j<N;j++){
int i=a[j].second;
ll x=sum1.sum(i+1,N);
x%=MOD;
ans+=A[i]*x%MOD;
ans%=MOD;
sum1.add(i+1,cnt1[i]);
}
//cout<<ans<<endl;
BIT tree2(n);
vector<ll> cnt2(N,0);
for(int i=0;i<N;i++){
int k=lower_bound(nums.begin(),nums.end(),A[i])-nums.begin();
tree2.add(k+1,1);
cnt2[i]=tree2.sum(k+2,n);
}
BIT sum2(N);
for(int j=N-1;j>=0;j--){
int i=a[j].second;
ll x=sum2.sum(i);
x%=MOD;
ans+=A[i]*x%MOD;
ans%=MOD;
sum2.add(i+1,cnt2[i]);
}
//cout<<ans<<endl;
for(int i=0;i<N;i++){
//cout<<cnt1[i]<<" "<<cnt2[i]<<endl;
ans+=(cnt1[i]*cnt2[i]%MOD)*A[i]%MOD;
ans%=MOD;
}
cout<<ans<<endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0