結果

問題 No.1733 Sum of Sorted Subarrays
ユーザー sapphire__15
提出日時 2021-10-20 15:32:51
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,487 bytes
コンパイル時間 1,679 ms
コンパイル使用メモリ 105,244 KB
最終ジャッジ日時 2025-01-25 01:59:55
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 5 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <atcoder/lazysegtree>
#include <atcoder/modint>
#include <functional>
#include <iostream>
#include <utility>
#include <vector>

using namespace std;

using mint = atcoder::modint998244353;

mint op(mint a, mint b) {
    return a + b;
}

mint e() {
    return mint(0);
}

mint mapping(mint f, mint x) {
    return f * x;
}

mint composition(mint f, mint g) {
    return f * g;
}

mint id() {
    return mint(1);
}

vector<size_t> sorted_idx(const vector<size_t> &A) {
    auto n = A.size();
    vector<pair<size_t, size_t>> tmp(n);
    for (size_t i = 0; i < n; i++) tmp[i] = {A[i], i};
    sort(begin(tmp), end(tmp));
    vector<size_t> ret(n);
    for (size_t i = 0; i < n; i++) ret[i] = tmp[i].second;
    return ret;
}

vector<mint> cumulative_sum(const vector<size_t> &A) {
    auto n = A.size();
    atcoder::lazy_segtree<mint, op, e, mint, mapping, composition, id> lst(vector<mint>(n, 1));
    vector<mint> ret(n);
    auto &&tmp = sorted_idx(A);
    for(auto &i : tmp) {
        ret[i] = lst.prod(0, i + 1) / lst.get(i);
        lst.apply(0, i+1, mint(2));
    }
    return ret;
}


int main() {
    size_t n;
    cin >> n;
    vector<size_t> A(n);
    for(size_t i = 0; i < n; i++) cin >> A[i];
    auto &&x = cumulative_sum(A);
    reverse(begin(A), end(A));
    auto &&y = cumulative_sum(A);
    reverse(begin(x), end(x));
    mint ans;
    for(size_t i = 0; i < n; i++) {
        ans += x[i] * y[i] * A[i];
    }
    cout << ans.val() << endl;
}
0