結果

問題 No.1300 Sum of Inversions
ユーザー nehan_der_thalnehan_der_thal
提出日時 2020-11-27 22:25:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 742 ms / 2,000 ms
コード長 949 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 87,108 KB
実行使用メモリ 178,036 KB
最終ジャッジ日時 2023-10-09 20:27:50
合計ジャッジ時間 21,718 ms
ジャッジサーバーID
(参考情報)
judge13 / judge11
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 65 ms
71,336 KB
testcase_01 AC 66 ms
71,484 KB
testcase_02 AC 63 ms
71,664 KB
testcase_03 AC 552 ms
137,028 KB
testcase_04 AC 518 ms
127,444 KB
testcase_05 AC 454 ms
118,380 KB
testcase_06 AC 603 ms
163,168 KB
testcase_07 AC 591 ms
143,200 KB
testcase_08 AC 686 ms
168,152 KB
testcase_09 AC 690 ms
167,840 KB
testcase_10 AC 385 ms
123,640 KB
testcase_11 AC 379 ms
122,628 KB
testcase_12 AC 527 ms
136,004 KB
testcase_13 AC 540 ms
127,636 KB
testcase_14 AC 721 ms
178,036 KB
testcase_15 AC 665 ms
167,888 KB
testcase_16 AC 566 ms
137,568 KB
testcase_17 AC 379 ms
121,332 KB
testcase_18 AC 419 ms
114,760 KB
testcase_19 AC 531 ms
122,624 KB
testcase_20 AC 607 ms
122,896 KB
testcase_21 AC 601 ms
123,176 KB
testcase_22 AC 564 ms
118,560 KB
testcase_23 AC 742 ms
162,996 KB
testcase_24 AC 487 ms
118,396 KB
testcase_25 AC 419 ms
116,828 KB
testcase_26 AC 393 ms
117,492 KB
testcase_27 AC 447 ms
118,048 KB
testcase_28 AC 673 ms
176,400 KB
testcase_29 AC 477 ms
123,392 KB
testcase_30 AC 667 ms
168,468 KB
testcase_31 AC 459 ms
118,120 KB
testcase_32 AC 469 ms
118,648 KB
testcase_33 AC 265 ms
108,004 KB
testcase_34 AC 278 ms
106,324 KB
testcase_35 AC 398 ms
142,616 KB
testcase_36 AC 463 ms
177,300 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class Bit:
    def __init__(self, n):
        self.size = n
        self.tree = [0] * (n + 1)

    def sum(self, i):
        s = 0
        while i > 0:
            s += self.tree[i]
            i -= i & -i
        return s

    def add(self, i, x):
        while i <= self.size:
            self.tree[i] += x
            i += i & -i

N, = map(int, input().split())

A = list(map(int, input().split()))
d = dict()
sA = sorted(list(set(A)))
la = len(sA)
for i, x in enumerate(sA):
    d[x]=i+1

bt = Bit(la)
bt2 = Bit(la)
X = [0]*N
for i, a in enumerate(A):
    X[i] = (bt.sum(la)-bt.sum(d[a]), bt2.sum(la)-bt2.sum(d[a]))
    bt.add(d[a], 1)
    bt2.add(d[a], a)
bt = Bit(la)
bt2 = Bit(la)
Y = [0]*N
for i in range(N-1, -1, -1):
    a = A[i]
    Y[i] = (bt.sum(d[a]-1), bt2.sum(d[a]-1))
    bt.add(d[a], 1)
    bt2.add(d[a], a)

MOD =  998244353
R = 0
for i in range(N):
    a, b = X[i]
    c, d = Y[i]
    R = (R + A[i]*a*c + b*c + d*a)%MOD
print(R)
0