結果
問題 | No.1300 Sum of Inversions |
ユーザー |
👑 |
提出日時 | 2022-07-04 15:05:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 935 ms / 2,000 ms |
コード長 | 1,416 bytes |
コンパイル時間 | 174 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 175,432 KB |
最終ジャッジ日時 | 2024-12-14 05:06:12 |
合計ジャッジ時間 | 25,882 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
MOD = 998244353class Bit:def __init__(self, n):self.size = nself.n0 = 1 << (n.bit_length() - 1)self.tree = [0] * (n + 1)def range_sum(self, l, r):return self.sum(r - 1) - self.sum(l - 1)def sum(self, i):i += 1s = 0while i > 0:s += self.tree[i]s %= MODi -= i & -ireturn sdef get(self, i):return self.sum(i) - self.sum(i - 1)def add(self, i, x):i += 1while i <= self.size:self.tree[i] += xself.tree[i] %= MODi += i & -idef lower_bound(self, x):pos = 0plus = self.n0while plus > 0:if pos + plus <= self.size and self.tree[pos + plus] < x:x -= self.tree[pos + plus]pos += plusplus //= 2return posn = int(input())A = list(map(int, input().split()))se = set(A)lst = sorted(se)dic = {a:i for i, a in enumerate(lst)}le = len(lst)bit1 = Bit(le)bit2 = Bit(le)cnt1 = Bit(le)cnt2 = Bit(le)ans = 0for a in A:b = dic[a]ans += bit2.range_sum(b + 1, le) + a * cnt2.range_sum(b + 1, le)ans %= MODbit2.add(b, bit1.range_sum(b + 1, le) + a * cnt1.range_sum(b + 1, le))cnt2.add(b, cnt1.range_sum(b + 1, le))bit1.add(b, a)cnt1.add(b, 1)print(ans)