結果

問題 No.1300 Sum of Inversions
ユーザー Ricky_ponRicky_pon
提出日時 2020-11-27 21:54:07
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 152 ms / 2,000 ms
コード長 4,820 bytes
コンパイル時間 2,565 ms
コンパイル使用メモリ 215,240 KB
実行使用メモリ 15,328 KB
最終ジャッジ日時 2024-07-26 12:24:36
合計ジャッジ時間 7,652 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 117 ms
12,512 KB
testcase_04 AC 113 ms
12,256 KB
testcase_05 AC 90 ms
10,776 KB
testcase_06 AC 129 ms
13,580 KB
testcase_07 AC 121 ms
13,268 KB
testcase_08 AC 141 ms
14,168 KB
testcase_09 AC 140 ms
14,060 KB
testcase_10 AC 74 ms
9,588 KB
testcase_11 AC 75 ms
9,724 KB
testcase_12 AC 114 ms
12,388 KB
testcase_13 AC 110 ms
12,100 KB
testcase_14 AC 152 ms
15,044 KB
testcase_15 AC 137 ms
14,148 KB
testcase_16 AC 120 ms
12,696 KB
testcase_17 AC 71 ms
9,300 KB
testcase_18 AC 83 ms
10,188 KB
testcase_19 AC 99 ms
11,364 KB
testcase_20 AC 102 ms
11,512 KB
testcase_21 AC 101 ms
11,512 KB
testcase_22 AC 91 ms
10,772 KB
testcase_23 AC 126 ms
13,580 KB
testcase_24 AC 94 ms
10,936 KB
testcase_25 AC 81 ms
10,024 KB
testcase_26 AC 78 ms
9,888 KB
testcase_27 AC 89 ms
10,748 KB
testcase_28 AC 142 ms
14,452 KB
testcase_29 AC 101 ms
11,500 KB
testcase_30 AC 137 ms
14,160 KB
testcase_31 AC 91 ms
10,908 KB
testcase_32 AC 96 ms
11,072 KB
testcase_33 AC 78 ms
15,208 KB
testcase_34 AC 92 ms
15,204 KB
testcase_35 AC 90 ms
15,328 KB
testcase_36 AC 94 ms
15,224 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

//#include <atcoder/all>
#define For(i, a, b) for (int(i) = (int)(a); (i) < (int)(b); ++(i))
#define rFor(i, a, b) for (int(i) = (int)(a)-1; (i) >= (int)(b); --(i))
#define rep(i, n) For((i), 0, (n))
#define rrep(i, n) rFor((i), (n), 0)
#define fi first
#define se second
using namespace std;
typedef long long lint;
typedef unsigned long long ulint;
typedef pair<int, int> pii;
typedef pair<lint, lint> pll;
template <class T>
bool chmax(T &a, const T &b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}
template <class T>
bool chmin(T &a, const T &b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template <class T>
T div_floor(T a, T b) {
    if (b < 0) a *= -1, b *= -1;
    return a >= 0 ? a / b : (a + 1) / b - 1;
}
template <class T>
T div_ceil(T a, T b) {
    if (b < 0) a *= -1, b *= -1;
    return a > 0 ? (a - 1) / b + 1 : a / b;
}

constexpr lint mod = 1000000007;
constexpr lint INF = mod * mod;
constexpr int MAX = 100010;

template <int_fast64_t MOD>
struct modint {
    using i64 = int_fast64_t;
    i64 a;
    modint(const i64 a_ = 0) : a(a_) {
        if (a > MOD)
            a %= MOD;
        else if (a < 0)
            (a %= MOD) += MOD;
    }
    modint inv() {
        i64 t = 1, n = MOD - 2, x = a;
        while (n) {
            if (n & 1) (t *= x) %= MOD;
            (x *= x) %= MOD;
            n >>= 1;
        }
        modint ret(t);
        return ret;
    }
    bool operator==(const modint x) const { return a == x.a; }
    bool operator!=(const modint x) const { return a != x.a; }
    modint operator+(const modint x) const { return modint(*this) += x; }
    modint operator-(const modint x) const { return modint(*this) -= x; }
    modint operator*(const modint x) const { return modint(*this) *= x; }
    modint operator/(const modint x) const { return modint(*this) /= x; }
    modint operator^(const lint x) const { return modint(*this) ^= x; }
    modint &operator+=(const modint &x) {
        a += x.a;
        if (a >= MOD) a -= MOD;
        return *this;
    }
    modint &operator-=(const modint &x) {
        a -= x.a;
        if (a < 0) a += MOD;
        return *this;
    }
    modint &operator*=(const modint &x) {
        (a *= x.a) %= MOD;
        return *this;
    }
    modint &operator/=(modint x) {
        (a *= x.inv().a) %= MOD;
        return *this;
    }
    modint &operator^=(lint n) {
        i64 ret = 1;
        while (n) {
            if (n & 1) (ret *= a) %= MOD;
            (a *= a) %= MOD;
            n >>= 1;
        }
        a = ret;
        return *this;
    }
    modint operator-() const { return modint(0) - *this; }
    modint &operator++() { return *this += 1; }
    modint &operator--() { return *this -= 1; }
    bool operator<(const modint x) const { return a < x.a; }
};

using mint = modint<998244353>;

vector<mint> fact;
vector<mint> revfact;

void setfact(int n) {
    fact.resize(n + 1);
    revfact.resize(n + 1);
    fact[0] = 1;
    rep(i, n) fact[i + 1] = fact[i] * mint(i + 1);

    revfact[n] = fact[n].inv();
    for (int i = n - 1; i >= 0; i--) revfact[i] = revfact[i + 1] * mint(i + 1);
}

mint getC(int n, int r) {
    if (n < r) return 0;
    return fact[n] * revfact[r] * revfact[n - r];
}

mint f(int n) { return mint(n) * mint(n - 1) / mint(2); }

template <typename T>
struct BinaryIndexedTree {
    vector<T> node;

    BinaryIndexedTree(int n) { node.resize(n + 1, {}); }

    void update(int i, T x) {
        ++i;
        while (i < (int)node.size()) {
            node[i] += x;
            i += (i & -i);
        }
    }

    T query(int i) {
        ++i;
        T ret = 0;
        while (i) {
            ret += node[i];
            i -= (i & -i);
        }
        return ret;
    }

    T query(int l, int r) { return query(r - 1) - query(l - 1); }
};

vector<int> v;

int get_idx(int x) { return lower_bound(v.begin(), v.end(), x) - v.begin(); }

int main() {
    int n;
    scanf("%d", &n);
    mint a[n];
    rep(i, n) {
        scanf("%lld", &a[i].a);
        v.push_back(a[i].a);
    }
    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()), v.end());

    int L[n], R[n];
    mint vl[n], vr[n];
    BinaryIndexedTree<int> btl(n), btr(n);
    BinaryIndexedTree<mint> bvl(n), bvr(n);
    rep(i, n) {
        int idx = get_idx(a[i].a);
        L[i] = btl.query(idx + 1, n);
        vl[i] = bvl.query(idx + 1, n);
        btl.update(idx, 1);
        bvl.update(idx, a[i]);
    }
    rrep(i, n) {
        int idx = get_idx(a[i].a);
        R[i] = btr.query(0, idx);
        vr[i] = bvr.query(0, idx);
        btr.update(idx, 1);
        bvr.update(idx, a[i]);
    }

    mint ans = 0;
    rep(i, n) { ans += a[i] * 1LL * L[i] * R[i] + vl[i] * R[i] + vr[i] * L[i]; }
    printf("%lld\n", ans.a);
}
0