結果

問題 No.1300 Sum of Inversions
ユーザー OmameBeansOmameBeans
提出日時 2020-11-29 10:15:35
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,125 ms / 2,000 ms
コード長 4,807 bytes
コンパイル時間 2,070 ms
コンパイル使用メモリ 186,272 KB
実行使用メモリ 50,080 KB
最終ジャッジ日時 2023-10-11 02:21:55
合計ジャッジ時間 28,649 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,356 KB
testcase_01 AC 1 ms
4,352 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 834 ms
39,212 KB
testcase_04 AC 801 ms
38,036 KB
testcase_05 AC 653 ms
32,280 KB
testcase_06 AC 957 ms
43,776 KB
testcase_07 AC 907 ms
41,724 KB
testcase_08 AC 1,020 ms
45,548 KB
testcase_09 AC 1,022 ms
45,656 KB
testcase_10 AC 512 ms
27,192 KB
testcase_11 AC 515 ms
27,584 KB
testcase_12 AC 803 ms
38,268 KB
testcase_13 AC 787 ms
37,712 KB
testcase_14 AC 1,125 ms
49,224 KB
testcase_15 AC 1,013 ms
45,396 KB
testcase_16 AC 863 ms
40,128 KB
testcase_17 AC 502 ms
26,680 KB
testcase_18 AC 588 ms
30,196 KB
testcase_19 AC 740 ms
34,400 KB
testcase_20 AC 738 ms
35,200 KB
testcase_21 AC 724 ms
34,924 KB
testcase_22 AC 646 ms
32,144 KB
testcase_23 AC 963 ms
43,424 KB
testcase_24 AC 679 ms
33,128 KB
testcase_25 AC 564 ms
29,044 KB
testcase_26 AC 552 ms
28,392 KB
testcase_27 AC 622 ms
31,420 KB
testcase_28 AC 1,040 ms
46,648 KB
testcase_29 AC 715 ms
34,844 KB
testcase_30 AC 1,016 ms
45,288 KB
testcase_31 AC 643 ms
32,532 KB
testcase_32 AC 676 ms
33,408 KB
testcase_33 AC 43 ms
12,280 KB
testcase_34 AC 98 ms
12,680 KB
testcase_35 AC 533 ms
50,080 KB
testcase_36 AC 555 ms
49,876 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define REP(i,n) for (int i = 0; i < (n); ++i)
template<class T> inline bool chmin(T& a, T b) {if (a > b) {a = b;return true;}return false;}
template<class T> inline bool chmax(T& a, T b) {if (a < b) {a = b;return true;}return false;}
using namespace std;
using ll = long long;
using P = pair<int,int>;
using Pl = pair<long long,long long>;
using veci = vector<int>;
using vecl = vector<long long>;
using vecveci = vector<vector<int>>;
using vecvecl = vector<vector<long long>>;
const int MOD = 998244353;
const double pi = acos(-1);
ll gcd(ll a, ll b) {if(b == 0) return a; else return gcd(b,a%b);}
ll lcm(ll a, ll b) {return a*b/gcd(a,b);}

template<int MOD> struct Fp {
    long long val;
    constexpr Fp(long long v = 0) noexcept : val(v % MOD) {
        if (val < 0) val += MOD;
    }
    constexpr int getmod() { return MOD; }
    constexpr Fp operator - () const noexcept {
        return val ? MOD - val : 0;
    }
    constexpr Fp operator + (const Fp& r) const noexcept { return Fp(*this) += r; }
    constexpr Fp operator - (const Fp& r) const noexcept { return Fp(*this) -= r; }
    constexpr Fp operator * (const Fp& r) const noexcept { return Fp(*this) *= r; }
    constexpr Fp operator / (const Fp& r) const noexcept { return Fp(*this) /= r; }
    constexpr Fp& operator += (const Fp& r) noexcept {
        val += r.val;
        if (val >= MOD) val -= MOD;
        return *this;
    }
    constexpr Fp& operator -= (const Fp& r) noexcept {
        val -= r.val;
        if (val < 0) val += MOD;
        return *this;
    }
    constexpr Fp& operator *= (const Fp& r) noexcept {
        val = val * r.val % MOD;
        return *this;
    }
    constexpr Fp& operator /= (const Fp& r) noexcept {
        long long a = r.val, b = MOD, u = 1, v = 0;
        while (b) {
            long long t = a / b;
            a -= t * b; swap(a, b);
            u -= t * v; swap(u, v);
        }
        val = val * u % MOD;
        if (val < 0) val += MOD;
        return *this;
    }
    constexpr bool operator == (const Fp& r) const noexcept {
        return this->val == r.val;
    }
    constexpr bool operator != (const Fp& r) const noexcept {
        return this->val != r.val;
    }
    friend constexpr ostream& operator << (ostream &os, const Fp<MOD>& x) noexcept {
        return os << x.val;
    }
    friend constexpr Fp<MOD> modpow(const Fp<MOD> &a, long long n) noexcept {
        if (n == 0) return 1;
        auto t = modpow(a, n / 2);
        t = t * t;
        if (n & 1) t = t * a;
        return t;
    }
};

template <typename T>
struct RSQ {

    //初期化
    const T INF = numeric_limits<T>::max();
    int n;         // 葉の数
    vector<T> dat; // 完全二分木の配列
    RSQ(int n_) : n(), dat(n_ * 4, 0) { // 葉の数は 2^x の形
        int x = 1;
        while (n_ > x) {
            x *= 2;
        }
        n = x;
    }

    //値の更新
    void update(int i, T x) {
        i += n-1;
        dat[i] += x;
        while(i > 0) {
            i = (i-1)/2;
            dat[i] += x;
        }
    }

    //任意区間の最小値
    // the minimum element of [a,b)
    T query(int a, int b) { return query_sub(a, b, 0, 0, n); }
    T query_sub(int a, int b, int k, int l, int r) {
        if(r <= a || b <= l) return 0;
        else if(a <= l && r <= b) return dat[k];
        else {
            T v1 = query_sub(a,b,2*k+1,l,(l+r)/2);
            T v2 = query_sub(a,b,2*k+2,(l+r)/2,r);
            return v1+v2;
        }
    }
};

using mint = Fp<MOD>;

int main() {
    int n; cin >> n;
    vecl a(n);
    REP(i,n) cin >> a[i];

    ll cnt = 0;
    map<ll,ll> mp;
    vecl b = a;
    sort(b.begin(),b.end());
    REP(i,n) {
        if(!mp.count(b[i])) {
            mp[b[i]] = cnt;
            cnt++;
        }
    }
    //REP(i,n) cout << mp[a[i]] << " ";
    //cout << endl;
    map<ll,ll> mp2;
    
    int m = mp.size();

    RSQ<ll> cntseg1(m), cntseg2(m);
    RSQ<mint> seg1(m), seg2(m);

    mint ans = 0;
    mint total = 0;

    vecl l(n), r(n);
    vector<mint> L(n), R(n);

    REP(i,n) {
        ll sum = cntseg1.query(mp[a[i]]+1, m);
        total = seg1.query(mp[a[i]]+1, m);
        l[i] = sum;
        L[i] = total;
        cntseg1.update(mp[a[i]],1);
        seg1.update(mp[a[i]], a[i]);
    }

    for(int i = n-1; i >= 0; i--) {
        ll sum = cntseg2.query(0,mp[a[i]]);
        total = seg2.query(0,mp[a[i]]);
        R[i] = total;
        r[i] = sum;
        cntseg2.update(mp[a[i]],1);
        seg2.update(mp[a[i]],a[i]);
    }

    REP(i,n) {
        mint x = L[i], y = R[i];
        //cout << i << ":" << l[i] << " "<< r[i] << endl;
        if(r[i] > 0 && l[i] > 0) {
            ans += (mint)l[i] * y + (mint)r[i] * x + (mint)a[i] * ((mint)l[i] * (mint)r[i]);
        }
    }

    cout << ans << endl;
}
0