結果

問題 No.1300 Sum of Inversions
ユーザー queeequeee
提出日時 2020-11-27 22:36:51
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 234 ms / 2,000 ms
コード長 3,168 bytes
コンパイル時間 1,226 ms
コンパイル使用メモリ 106,228 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2024-07-26 18:56:00
合計ジャッジ時間 8,177 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 172 ms
10,496 KB
testcase_04 AC 167 ms
10,240 KB
testcase_05 AC 137 ms
9,088 KB
testcase_06 AC 199 ms
11,264 KB
testcase_07 AC 192 ms
11,136 KB
testcase_08 AC 214 ms
11,776 KB
testcase_09 AC 217 ms
11,648 KB
testcase_10 AC 111 ms
8,192 KB
testcase_11 AC 113 ms
8,192 KB
testcase_12 AC 178 ms
10,240 KB
testcase_13 AC 167 ms
10,112 KB
testcase_14 AC 234 ms
12,544 KB
testcase_15 AC 219 ms
11,648 KB
testcase_16 AC 180 ms
10,624 KB
testcase_17 AC 109 ms
7,936 KB
testcase_18 AC 125 ms
8,576 KB
testcase_19 AC 149 ms
9,600 KB
testcase_20 AC 153 ms
9,600 KB
testcase_21 AC 152 ms
9,600 KB
testcase_22 AC 138 ms
9,216 KB
testcase_23 AC 198 ms
11,264 KB
testcase_24 AC 141 ms
9,216 KB
testcase_25 AC 120 ms
8,448 KB
testcase_26 AC 116 ms
8,320 KB
testcase_27 AC 132 ms
8,832 KB
testcase_28 AC 221 ms
11,904 KB
testcase_29 AC 149 ms
9,600 KB
testcase_30 AC 212 ms
11,648 KB
testcase_31 AC 141 ms
9,216 KB
testcase_32 AC 141 ms
9,344 KB
testcase_33 AC 118 ms
12,544 KB
testcase_34 AC 168 ms
12,672 KB
testcase_35 AC 133 ms
12,544 KB
testcase_36 AC 153 ms
12,544 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

//g++ t.cpp -o t && t < p.txt
//d>p.txt&&t<p.txt&&t2<p.txt

#include <iostream> // プログラムの耳と口
#include <vector> // 超有能配列秘書
#include <algorithm> // みんな大好きソートと二分探索
#include <queue> // きゅーちゃん、だいすき
#include <string> // to_string
#include <functional> // 関数を変数に入れる子
#include <set> // 値の取得・挿入・削除を高速に
#include <map> // setの妹 これまた優秀
#include <random> // 乱択さん
#include <ctime> // へびになりたい
#include <bitset> // へびになった ｼｬｰ
#include <cassert>

// #include <atcoder/all>
// using namespace atcoder; // コンパイル重い
// g++ t.cpp -o t -I .

#define DB cerr<<"D"<<endl

using namespace std; using ll=long long; using ld=long double; const int INF=1e9; const ll LINF=1e18; const double dINF = 1e18; const ld ldINF = 1e18; const double EPS = 1e-6;
using P=pair<ll,ll>;
const ll M = 998244353;
ll mod_pow(ll x, ll a) { ll an = 1; while(a > 0) { if (a&1) an = an * x % M; x = x * x % M; a >>= 1;} return an;}
ll mod_pow(ll x, ll a, ll m) { ll an = 1; while(a > 0) { if (a&1) an = an * x % m; x = x * x % m; a >>= 1;} return an;}
void add(ll& x, ll y) {x+=y; x%=M;}; void mul(ll& x, ll y) {x*=y; x%=M;}; template<typename T, typename U> void chmax(T& x, U y) {if (x<y) x=y;}; template<typename T, typename U> void chmin(T& x, U y) {if (x>y) x=y;} bool vaild(int x, int y, int hh, int ww){return 0<=x&&x<hh&&0<=y&&y<ww;}
ll gcd(ll a, ll b) {if (b==0) return a; else return gcd(b, a%b);}
int keta(ll a) {int res=0; while(a>0) res+=a%10, a/=10; return res;}
const int up[]={1,-1,0,0}, lf[]={0,0,1,-1};

struct BIT { // 0-index
  int n; vector<ll> v;
  BIT(int m) : n(m) {v.resize(m);}
  ll sum(int i) {
    //cout<<"sum "<<i<<" "<<n<<" ";
    if (i<0 || i>=n) return 0;
    ll res=0; for(; i>=0; i=(i&(i+1))-1) res+=v[i]; res%=M; return res;
  }
  ll sum(int a, int b) {
    return (sum(b) - sum(a-1) + M )% M ;
  }
  void add(int i, ll x) {for(;i<n;i|=i+1) v[i]+=x,v[i]%=M;}
  void print() {
    for(int i=0;i<n;i++) cout<<sum(i)-sum(i-1)<<" "; cout<<endl;
  }
};

int main() {
  int n; cin>>n;
  ll a[n]; for(int i=0;i<n;i++) cin>>a[i];
  ll b[n]; copy(a,a+n,b); sort(b,b+n);
  BIT b1(n),b2(n),c1(n),c2(n);
  for(int i=0;i<n;i++) {
    int id = lower_bound(b,b+n,a[i])-b;
    b2.add(id,1); c2.add(id,a[i]);
  }
  ll an=0;
  for(int i=0;i<n;i++) {
    int id = lower_bound(b,b+n,a[i])-b;
    b1.add(id,1);
    c1.add(id,a[i]);
    b2.add(id,-1);
    c2.add(id,-a[i]);

    ll nu = b2.sum(0,id-1) * b1.sum(id+1,n-1) % M;

    //cout << "@ id " << id << " a " << a[i] << " " << nu << " " << c2.sum(0,id-1) << " " << c1.sum(id+1,n-1) << endl;
    //b1.print(); b2.print(); c1.print(); c2.print();

    an += a[i] * nu % M;
    an %= M;
    an += (b1.sum(id+1,n-1) * c2.sum(0,id-1) % M  + b2.sum(0,id-1) * c1.sum(id+1,n-1) % M) % M;
    an %= M;
  }
  cout<<an<<endl;
}

// ・配列の大きさok？　・priority_queueはgreater？　・debug消した？
// ・落ち着いて。提出まで4分待ってね……WJ……1/10……2/10……

// Thank you for making problems and running the contest
0