結果

問題 No.2616 中央番目の中央値
ユーザー 👑 p-adic
提出日時 2023-12-19 18:48:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 696 ms / 2,000 ms
コード長 633 bytes
コンパイル時間 367 ms
コンパイル使用メモリ 82,392 KB
実行使用メモリ 154,116 KB
最終ジャッジ日時 2024-09-27 11:13:54
合計ジャッジ時間 13,342 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

I,R=input,range
N=int(I())
X=list(map(int,I().split()))
a=0
P=998244353
e=[0,1]
for i in R(2,N+1):e+=[P-P//i*e[P%i]%P]
f=[1]
for i in R(N):f+=[f[-1]*(i+1)%P]
g=[1]
for i in R(N):g+=[g[-1]*e[i+1]%P]
def B(i,t,F):
	i+=1
	while i<=N:F[i],i=F[i]+t,i+(i&-i)
def M(r,F):
	a=0
	i=min(r+1,N)
	while i:a,i=a+F[i],i-(i&-i)
	return a
F=[[0]*(N+1)for i in R(2)]
def A(l,r,t):
	B(l,-(l-1)*t,F[0]),B(r+1,r*t,F[0]),B(l,t,F[1]),B(r+1,-t,F[1])
def L(r):
	return M(r,F[0])+r*M(r,F[1])
def S(l,r):
	return L(r)-L(l-1)
for i in R(N):
	x=S(X[i]-1,X[i]-1)
	a+=f[2*x+N-X[i]-i]*g[x]*g[x+N-X[i]-i]*f[X[i]-1+i-2*x]*g[i-x]*g[X[i]-1-x]
	A(X[i],N-1,1)
print(a%P)
0