結果
問題 |
No.1996 <><
|
ユーザー |
![]() |
提出日時 | 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)