結果

問題 No.1300 Sum of Inversions
ユーザー donutholedonuthole
提出日時 2020-11-28 16:51:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,001 ms / 2,000 ms
コード長 1,653 bytes
コンパイル時間 667 ms
コンパイル使用メモリ 86,992 KB
実行使用メモリ 209,144 KB
最終ジャッジ日時 2023-10-10 00:44:10
合計ジャッジ時間 30,390 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 279 ms
93,708 KB
testcase_01 AC 277 ms
93,392 KB
testcase_02 AC 274 ms
93,692 KB
testcase_03 AC 852 ms
181,592 KB
testcase_04 AC 822 ms
181,248 KB
testcase_05 AC 734 ms
163,012 KB
testcase_06 AC 930 ms
192,844 KB
testcase_07 AC 899 ms
191,752 KB
testcase_08 AC 956 ms
198,092 KB
testcase_09 AC 960 ms
197,932 KB
testcase_10 AC 656 ms
154,188 KB
testcase_11 AC 654 ms
153,876 KB
testcase_12 AC 838 ms
181,320 KB
testcase_13 AC 832 ms
180,192 KB
testcase_14 AC 1,001 ms
209,144 KB
testcase_15 AC 965 ms
197,588 KB
testcase_16 AC 876 ms
182,892 KB
testcase_17 AC 642 ms
148,596 KB
testcase_18 AC 696 ms
155,964 KB
testcase_19 AC 779 ms
171,000 KB
testcase_20 AC 789 ms
171,348 KB
testcase_21 AC 767 ms
171,412 KB
testcase_22 AC 737 ms
163,116 KB
testcase_23 AC 912 ms
192,924 KB
testcase_24 AC 745 ms
163,384 KB
testcase_25 AC 680 ms
155,196 KB
testcase_26 AC 669 ms
154,748 KB
testcase_27 AC 721 ms
162,824 KB
testcase_28 AC 981 ms
207,784 KB
testcase_29 AC 771 ms
171,272 KB
testcase_30 AC 952 ms
198,200 KB
testcase_31 AC 737 ms
162,980 KB
testcase_32 AC 746 ms
164,164 KB
testcase_33 AC 545 ms
133,820 KB
testcase_34 AC 546 ms
138,044 KB
testcase_35 AC 630 ms
206,184 KB
testcase_36 AC 671 ms
208,556 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import typing
import sys
import math
import collections
import bisect
import itertools
import heapq
import decimal
import copy

# sys.setrecursionlimit(10000001)
INF = 10 ** 20
# MOD = 10 ** 9 + 7
MOD = 998244353


def ni(): return int(sys.stdin.buffer.readline())
def ns(): return map(int, sys.stdin.buffer.readline().split())
def na(): return list(map(int, sys.stdin.buffer.readline().split()))
def na1(): return list(map(lambda x: int(x)-1, sys.stdin.buffer.readline().split()))


# ===CODE===
# https://ikatakos.com/pot/programming_algorithm/dynamic_programming/inversion
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


def main():
    n = ni()
    a = na()

    sa = sorted(list(set(a)))
    d = {ai: i+1 for i, ai in enumerate(sa)}

    cnt = Bit(n)
    val = Bit(n)

    cntRes = [0]*n
    valRes = [0]*n
    for i, ai in enumerate(a):
        idx = d[ai]
        cntRes[i] = cnt.sum(n)-cnt.sum(idx)
        valRes[i] = val.sum(n)-val.sum(idx)
        cnt.add(idx, 1)
        val.add(idx, ai)

    ans = 0
    cnt = Bit(n)
    val = Bit(n)
    for i in range(n-1, -1, -1):
        idx = d[a[i]]
        c = cnt.sum(idx-1)
        v = val.sum(idx-1)
        cnt.add(idx,1)
        val.add(idx,a[i])

        ans += a[i]*cntRes[i]*c
        ans += c*valRes[i]
        ans += v*cntRes[i]
        ans %= MOD
    print(ans)


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