結果

問題 No.1300 Sum of Inversions
ユーザー wolgnikwolgnik
提出日時 2020-11-27 21:58:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,074 ms / 2,000 ms
コード長 1,158 bytes
コンパイル時間 548 ms
コンパイル使用メモリ 86,696 KB
実行使用メモリ 112,408 KB
最終ジャッジ日時 2023-10-01 05:42:29
合計ジャッジ時間 30,192 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 78 ms
71,132 KB
testcase_01 AC 81 ms
71,052 KB
testcase_02 AC 87 ms
71,164 KB
testcase_03 AC 896 ms
103,852 KB
testcase_04 AC 848 ms
104,092 KB
testcase_05 AC 699 ms
99,836 KB
testcase_06 AC 952 ms
105,824 KB
testcase_07 AC 901 ms
106,624 KB
testcase_08 AC 982 ms
106,796 KB
testcase_09 AC 978 ms
106,096 KB
testcase_10 AC 568 ms
96,964 KB
testcase_11 AC 580 ms
96,896 KB
testcase_12 AC 818 ms
103,988 KB
testcase_13 AC 793 ms
103,900 KB
testcase_14 AC 1,074 ms
109,812 KB
testcase_15 AC 971 ms
106,392 KB
testcase_16 AC 847 ms
103,580 KB
testcase_17 AC 545 ms
94,968 KB
testcase_18 AC 618 ms
97,404 KB
testcase_19 AC 725 ms
101,812 KB
testcase_20 AC 743 ms
101,576 KB
testcase_21 AC 749 ms
101,648 KB
testcase_22 AC 674 ms
99,832 KB
testcase_23 AC 924 ms
105,580 KB
testcase_24 AC 704 ms
99,916 KB
testcase_25 AC 632 ms
96,984 KB
testcase_26 AC 610 ms
97,104 KB
testcase_27 AC 667 ms
100,088 KB
testcase_28 AC 1,007 ms
108,648 KB
testcase_29 AC 730 ms
101,704 KB
testcase_30 AC 988 ms
106,620 KB
testcase_31 AC 680 ms
99,920 KB
testcase_32 AC 699 ms
99,856 KB
testcase_33 AC 367 ms
112,408 KB
testcase_34 AC 379 ms
111,968 KB
testcase_35 AC 368 ms
111,032 KB
testcase_36 AC 429 ms
110,292 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
N = int(input())
a = list(map(int, input().split()))
mod = 998244353

class FenwickTree:
  def __init__(self, n):
    self.n = n
    self.data = [0] * (n + 2)
  def sum(self, l, r):
    s = 0
    while l > 0:
      s -= self.data[l]
      l -= l & -l
    while r > 0:
      s += self.data[r]
      r -= r & -r
    return s
  def add(self, i, x):
    i += 1
    while i <= self.n:
      self.data[i] += x
      i += i & -i
  def lowerbound(self, s):
    x = 0
    y = 0
    for i in range(self.n.bit_length(), -1, -1):
      k = x + (1 << i)
      if k <= self.n and (y + self.data[k] < s):
        y += self.data[k]
        x += 1 << i
    return x + 1

fwk = FenwickTree(N)
fwk2 = FenwickTree(N)
sa = [(a[i], i) for i in range(N)]
sa.sort(reverse = True)
table = [0] * N
table2 = [0] * N
for x, i in sa:
  table[i] = fwk.sum(0, i) % mod
  table2[i] = fwk2.sum(0, i)
  fwk.add(i, x)
  fwk2.add(i, 1)
#print(table, table2)
fwk = FenwickTree(N)
fwk2 = FenwickTree(N)
res = 0
for x, i in sa:
  res += fwk.sum(0, i) + fwk2.sum(0, i) * x
  res %= mod
  fwk.add(i, table[i] + table2[i] * x)
  fwk2.add(i, table2[i])
print(res)
0