結果
問題 | No.1300 Sum of Inversions |
ユーザー |
👑 |
提出日時 | 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 |
ソースコード
#include <iostream>#include <vector>#include <map>#include <algorithm>using namespace std;const long long MOD = 998244353;//Btemplate<typename T>struct BIT{//1_Indexedint 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);}};//Eint 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;}