結果

問題 No.2250 Split Permutation
ユーザー kcvlex
提出日時 2023-03-17 23:10:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,046 bytes
コンパイル時間 2,988 ms
コンパイル使用メモリ 151,368 KB
最終ジャッジ日時 2025-02-11 14:15:56
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 11 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <atcoder/all>

#define ALL(V) std::begin(V), std::end(V)

using ll = long long;
using modint = atcoder::modint998244353;

int main() {
    int N;
    std::cin >> N;
    std::vector<int> P(N), rP(N);
    for (int i = 0; i < N; i++) {
        std::cin >> P[i];
        P[i]--;
        rP[P[i]] = i;
    }

    std::vector<modint> pow2(N + 10);
    pow2[0] = 1;
    for (int i = 0; i < N + 9; i++) pow2[i + 1] = pow2[i] * 2;

    modint ans = 0;
    {
        atcoder::fenwick_tree<modint> cnt(N);
        for (int i = 0; i < N; i++) {
            int idx = rP[i];
            auto sum = cnt.sum(idx, N);
            ans += sum;
            cnt.add(idx, 1);
        }
        ans *= pow2[N - 1];
    }
    
    atcoder::fenwick_tree<modint> ft(N);
    for (int i = 0; i < N; i++) {
        int idx = rP[i];
        auto sum = ft.sum(idx, N);
        int p = N - 1 + idx;
        ans -= sum * pow2[p];
        ft.add(idx, pow2[idx].inv());
    }
    std::cout << ans.val() << std::endl;
    return 0;
}
0