結果

問題 No.2250 Split Permutation
ユーザー bayashikobayashiko
提出日時 2024-09-18 13:36:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 471 ms / 3,000 ms
コード長 1,579 bytes
コンパイル時間 182 ms
コンパイル使用メモリ 81,744 KB
実行使用メモリ 119,136 KB
最終ジャッジ日時 2024-09-18 13:36:22
合計ジャッジ時間 7,482 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,500 KB
testcase_01 AC 39 ms
52,464 KB
testcase_02 AC 39 ms
53,248 KB
testcase_03 AC 39 ms
52,308 KB
testcase_04 AC 38 ms
52,552 KB
testcase_05 AC 39 ms
52,856 KB
testcase_06 AC 39 ms
52,008 KB
testcase_07 AC 40 ms
53,208 KB
testcase_08 AC 39 ms
52,684 KB
testcase_09 AC 39 ms
53,224 KB
testcase_10 AC 39 ms
53,000 KB
testcase_11 AC 39 ms
52,668 KB
testcase_12 AC 39 ms
52,696 KB
testcase_13 AC 38 ms
52,320 KB
testcase_14 AC 40 ms
52,016 KB
testcase_15 AC 39 ms
52,948 KB
testcase_16 AC 39 ms
52,912 KB
testcase_17 AC 39 ms
52,744 KB
testcase_18 AC 453 ms
118,824 KB
testcase_19 AC 402 ms
113,988 KB
testcase_20 AC 357 ms
109,196 KB
testcase_21 AC 270 ms
95,932 KB
testcase_22 AC 332 ms
106,072 KB
testcase_23 AC 120 ms
80,512 KB
testcase_24 AC 338 ms
106,204 KB
testcase_25 AC 471 ms
118,524 KB
testcase_26 AC 197 ms
90,500 KB
testcase_27 AC 99 ms
78,100 KB
testcase_28 AC 130 ms
82,540 KB
testcase_29 AC 424 ms
119,136 KB
testcase_30 AC 162 ms
82,948 KB
testcase_31 AC 83 ms
76,432 KB
testcase_32 AC 150 ms
85,612 KB
testcase_33 AC 267 ms
98,900 KB
testcase_34 AC 116 ms
80,904 KB
testcase_35 AC 206 ms
91,396 KB
testcase_36 AC 149 ms
84,796 KB
testcase_37 AC 455 ms
118,576 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import sys
import sys
def main():
    import sys
    import sys
    sys.setrecursionlimit(1 << 25)
    MOD = 998244353

    class BIT:
        def __init__(self, size):
            self.N = size
            self.tree = [0] * (self.N + 2)

        def update(self, idx, val):
            while idx <= self.N:
                self.tree[idx] = (self.tree[idx] + val) % MOD
                idx += idx & -idx

        def query(self, idx):
            res = 0
            while idx > 0:
                res = (res + self.tree[idx]) % MOD
                idx -= idx & -idx
            return res

        def range_query(self, l, r):
            if l > r:
                return 0
            return (self.query(r) - self.query(l -1)) % MOD

    input = sys.stdin.read
    data = input().split()
    N = int(data[0])
    P = list(map(int, data[1:N+1]))

    pow2 = [1] * (N+1)
    for k in range(1, N+1):
        pow2[k] = (pow2[k-1] *2) % MOD

    # Compute K: inversion count
    bit_inv = BIT(N)
    K =0
    for j in range(N):
        val = P[j]
        cnt = bit_inv.range_query(val +1, N)
        K = (K + cnt) % MOD
        bit_inv.update(val,1)

    # Compute sum_T
    bit_sum_T = BIT(N)
    sum_T =0
    for j in range(N):
        val = P[j]
        sum_i = bit_sum_T.range_query(val +1, N)
        term_j = (sum_i * pow2[N -j -1]) % MOD
        sum_T = (sum_T + term_j) % MOD
        bit_sum_T.update(val, pow2[j])
    # Compute answer
    ans = (K * pow2[N-1] - sum_T) % MOD
    if ans <0:
        ans += MOD
    print(ans)

if __name__ == "__main__":
    main()
0