結果
| 問題 |
No.1996 <><
|
| コンテスト | |
| ユーザー |
MasKoaTS
|
| 提出日時 | 2021-12-24 15:19:39 |
| 言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
| 結果 |
RE
|
| 実行時間 | - |
| コード長 | 874 bytes |
| コンパイル時間 | 119 ms |
| コンパイル使用メモリ | 12,544 KB |
| 実行使用メモリ | 10,752 KB |
| 最終ジャッジ日時 | 2024-11-26 02:51:23 |
| 合計ジャッジ時間 | 1,874 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | RE * 4 |
| other | RE * 29 |
ソースコード
import sys
input = sys.stdin.readline
MOD = 998244353
class BIT:
def __init__(self, n):
self.size = n
self.bit = [0] * (n + 1)
def add(self, k, p):
t = k + 1
while(t <= self.size):
self.bit[t] += p
self.bit[t] %= MOD
t += t & -t
def sum(self, r):
res, t = 0, r
while(t > 0):
res += self.bit[t]
res %= MOD
t -= t & -t
return res
"""
Main Code
"""
n = int(input())
a = list(map(int,input().split()))
s = sorted(set(a))
d = {}
for i in range(len(s)):
d[s[i]] = i
for i in range(n):
a[i] = d[a[i]]
b = [len(s) - 1 - i for i in a]
lbt = BIT(n); lcnt = [0] * n
rbt = BIT(n); rcnt = [0] * n
for i in range(n):
lbt.add(a[i],1)
rbt.add(b[n - 1 - i],1)
lcnt[i] = lbt.sum(a[i])
rcnt[n - 1 - i] = rbt.sum(b[n - 1 - i])
bt = BIT(n)
ans = 0
for i in range(n):
bt.add(b[i],lcnt[i])
ans += bt.sum(b[i]) * rcnt[i]
ans %= MOD
print(ans)
MasKoaTS