結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0