結果

問題 No.1300 Sum of Inversions
ユーザー naniwazunaniwazu
提出日時 2020-11-27 23:04:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 666 ms / 2,000 ms
コード長 934 bytes
コンパイル時間 267 ms
コンパイル使用メモリ 87,356 KB
実行使用メモリ 181,648 KB
最終ジャッジ日時 2023-10-09 21:01:40
合計ジャッジ時間 17,374 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 59 ms
71,296 KB
testcase_01 AC 62 ms
71,620 KB
testcase_02 AC 63 ms
71,172 KB
testcase_03 AC 498 ms
138,652 KB
testcase_04 AC 469 ms
126,348 KB
testcase_05 AC 400 ms
117,308 KB
testcase_06 AC 526 ms
167,028 KB
testcase_07 AC 503 ms
145,224 KB
testcase_08 AC 572 ms
172,048 KB
testcase_09 AC 576 ms
172,232 KB
testcase_10 AC 338 ms
114,284 KB
testcase_11 AC 370 ms
114,224 KB
testcase_12 AC 476 ms
126,372 KB
testcase_13 AC 489 ms
126,524 KB
testcase_14 AC 666 ms
181,648 KB
testcase_15 AC 608 ms
172,036 KB
testcase_16 AC 496 ms
139,248 KB
testcase_17 AC 329 ms
110,432 KB
testcase_18 AC 361 ms
114,064 KB
testcase_19 AC 427 ms
121,352 KB
testcase_20 AC 415 ms
121,600 KB
testcase_21 AC 432 ms
121,620 KB
testcase_22 AC 452 ms
117,304 KB
testcase_23 AC 550 ms
167,128 KB
testcase_24 AC 404 ms
117,336 KB
testcase_25 AC 345 ms
114,780 KB
testcase_26 AC 358 ms
114,692 KB
testcase_27 AC 359 ms
117,372 KB
testcase_28 AC 624 ms
180,260 KB
testcase_29 AC 448 ms
121,572 KB
testcase_30 AC 579 ms
172,620 KB
testcase_31 AC 399 ms
117,324 KB
testcase_32 AC 416 ms
117,652 KB
testcase_33 AC 232 ms
112,832 KB
testcase_34 AC 254 ms
110,424 KB
testcase_35 AC 340 ms
141,372 KB
testcase_36 AC 345 ms
180,936 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