結果

問題 No.1300 Sum of Inversions
ユーザー naniwazunaniwazu
提出日時 2020-11-27 23:04:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 728 ms / 2,000 ms
コード長 934 bytes
コンパイル時間 631 ms
コンパイル使用メモリ 82,576 KB
実行使用メモリ 181,432 KB
最終ジャッジ日時 2024-07-26 19:26:45
合計ジャッジ時間 19,090 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,724 KB
testcase_01 AC 37 ms
52,560 KB
testcase_02 AC 37 ms
53,260 KB
testcase_03 AC 569 ms
127,008 KB
testcase_04 AC 549 ms
126,968 KB
testcase_05 AC 445 ms
118,972 KB
testcase_06 AC 610 ms
132,272 KB
testcase_07 AC 585 ms
132,320 KB
testcase_08 AC 658 ms
150,060 KB
testcase_09 AC 645 ms
150,032 KB
testcase_10 AC 378 ms
123,936 KB
testcase_11 AC 379 ms
124,484 KB
testcase_12 AC 550 ms
127,004 KB
testcase_13 AC 536 ms
127,060 KB
testcase_14 AC 728 ms
181,432 KB
testcase_15 AC 646 ms
149,836 KB
testcase_16 AC 548 ms
127,200 KB
testcase_17 AC 359 ms
123,628 KB
testcase_18 AC 400 ms
115,056 KB
testcase_19 AC 486 ms
122,572 KB
testcase_20 AC 491 ms
122,900 KB
testcase_21 AC 481 ms
122,812 KB
testcase_22 AC 439 ms
118,976 KB
testcase_23 AC 613 ms
132,376 KB
testcase_24 AC 455 ms
119,540 KB
testcase_25 AC 392 ms
118,160 KB
testcase_26 AC 387 ms
118,576 KB
testcase_27 AC 436 ms
118,440 KB
testcase_28 AC 681 ms
156,524 KB
testcase_29 AC 494 ms
123,468 KB
testcase_30 AC 650 ms
150,192 KB
testcase_31 AC 447 ms
118,996 KB
testcase_32 AC 457 ms
119,628 KB
testcase_33 AC 255 ms
111,960 KB
testcase_34 AC 271 ms
107,496 KB
testcase_35 AC 392 ms
141,988 KB
testcase_36 AC 414 ms
180,724 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
def BIT_query(BIT,idx):
    res_sum = 0
    while idx > 0:
        res_sum += BIT[idx]
        idx -= idx&(-idx)
    return res_sum % mod

def BIT_update(BIT,idx,x):
    while idx <= N:
        BIT[idx] += x
        BIT[idx] %= mod
        idx += idx&(-idx)
    return

def compress(li):
    *XS, = set(li)
    XS.sort()
    return {e: i for i, e in enumerate(XS)}

N = int(input())
A = list(map(int,input().split()))[::-1]
cA = compress(A)
ans = 0
BIT1 = [0]*(N+1)
BIT2 = [0]*(N+1)
BIT3 = [0]*(N+1)
BIT4 = [0]*(N+1)
for i in range(N):
    c = cA[A[i]]+1
    if c > 1:
        ans += BIT_query(BIT4,c-1)
        ans += BIT_query(BIT3,c-1)*A[i]
        ans %= mod
    BIT_update(BIT1,c,1)
    BIT_update(BIT2,c,A[i])
    num = 0
    SUM = 0
    if c > 1:
        num = BIT_query(BIT1,c-1)
        SUM = BIT_query(BIT2,c-1)
    BIT_update(BIT3,c,num)
    BIT_update(BIT4,c,A[i]*num)
    BIT_update(BIT4,c,SUM)
print(ans)
0