結果
問題 | No.2616 中央番目の中央値 |
ユーザー | とりゐ |
提出日時 | 2024-01-26 21:40:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 658 ms / 2,000 ms |
コード長 | 3,500 bytes |
コンパイル時間 | 468 ms |
コンパイル使用メモリ | 81,572 KB |
実行使用メモリ | 136,980 KB |
最終ジャッジ日時 | 2024-01-26 21:41:02 |
合計ジャッジ時間 | 13,288 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 49 ms
64,148 KB |
testcase_01 | AC | 51 ms
64,148 KB |
testcase_02 | AC | 50 ms
64,148 KB |
testcase_03 | AC | 53 ms
64,148 KB |
testcase_04 | AC | 50 ms
64,148 KB |
testcase_05 | AC | 51 ms
64,148 KB |
testcase_06 | AC | 51 ms
64,148 KB |
testcase_07 | AC | 50 ms
64,148 KB |
testcase_08 | AC | 50 ms
64,148 KB |
testcase_09 | AC | 59 ms
66,396 KB |
testcase_10 | AC | 65 ms
70,944 KB |
testcase_11 | AC | 67 ms
73,016 KB |
testcase_12 | AC | 78 ms
77,660 KB |
testcase_13 | AC | 112 ms
81,392 KB |
testcase_14 | AC | 114 ms
81,496 KB |
testcase_15 | AC | 115 ms
81,216 KB |
testcase_16 | AC | 121 ms
82,392 KB |
testcase_17 | AC | 132 ms
84,320 KB |
testcase_18 | AC | 171 ms
88,084 KB |
testcase_19 | AC | 248 ms
97,704 KB |
testcase_20 | AC | 231 ms
97,704 KB |
testcase_21 | AC | 395 ms
117,924 KB |
testcase_22 | AC | 585 ms
136,812 KB |
testcase_23 | AC | 588 ms
136,468 KB |
testcase_24 | AC | 465 ms
136,980 KB |
testcase_25 | AC | 395 ms
136,464 KB |
testcase_26 | AC | 605 ms
136,976 KB |
testcase_27 | AC | 584 ms
136,464 KB |
testcase_28 | AC | 607 ms
136,468 KB |
testcase_29 | AC | 559 ms
136,460 KB |
testcase_30 | AC | 559 ms
136,460 KB |
testcase_31 | AC | 639 ms
136,844 KB |
testcase_32 | AC | 559 ms
136,468 KB |
testcase_33 | AC | 587 ms
136,464 KB |
testcase_34 | AC | 626 ms
136,976 KB |
testcase_35 | AC | 658 ms
136,976 KB |
testcase_36 | AC | 560 ms
136,464 KB |
ソースコード
from sys import stdin input=lambda :stdin.readline()[:-1] class segtree(): def __init__(self,init,func,ide): self.n=len(init) self.func=func self.ide=ide self.size=1<<(self.n-1).bit_length() self.tree=[self.ide for i in range(2*self.size)] for i in range(self.n): self.tree[self.size+i]=init[i] for i in range(self.size-1,0,-1): self.tree[i]=self.func(self.tree[2*i], self.tree[2*i|1]) def update(self,k,x): k+=self.size self.tree[k]=x k>>=1 while k: self.tree[k]=self.func(self.tree[2*k],self.tree[k*2|1]) k>>=1 def get(self,i): return self.tree[i+self.size] def query(self,l,r): l+=self.size r+=self.size l_res=self.ide r_res=self.ide while l<r: if l&1: l_res=self.func(l_res,self.tree[l]) l+=1 if r&1: r-=1 r_res=self.func(self.tree[r],r_res) l>>=1 r>>=1 return self.func(l_res,r_res) def max_right(self,l,f): assert 0<=l<=self.n assert f(self.ide) if l==self.n: return self.n l+=self.size res=self.ide while True: while l&1==0: l>>=1 if not f(self.func(res,self.tree[l])): while l<self.size: l*=2 if f(self.func(res,self.tree[l])): res=self.func(res,self.tree[l]) l+=1 return l-self.size res=self.func(res,self.tree[l]) l+=1 if l&(-l)==l: break return self.n def min_left(self,r,f): assert 0<=r<=self.n assert f(self.ide) if r==0: return 0 r+=self.size res=self.ide while True: r-=1 while r>1 and r&1: r>>=1 if not f(self.func(self.tree[r],res)): while r<self.size: r=2*r+1 if f(self.func(self.tree[r],res)): res=self.func(self.tree[r],res) r-=1 return r+1-self.size res=self.func(self.tree[r],res) if r&(-r)==r: break return 0 def __getitem__(self,i): return self.get(i) def __setitem__(self,key,val): self.update(key,val) def __iter__(self): for i in range(self.n): yield self.tree[i+self.size] def __str__(self): return str(list(self)) mod=998244353 table_size=3*10**5 fac=[1]*(table_size+1) finv=[1]*(table_size+1) for i in range(2,table_size+1): fac[i]=fac[i-1]*i%mod finv[table_size]=pow(fac[table_size],mod-2,mod) for i in range(table_size-1,-1,-1): finv[i]=finv[i+1]*(i+1)%mod def rebuild(n): global table_size,fac,finv fac+=[0]*(n-table_size) fac+=[0]*(n-table_size) finv+=[0]*(n-table_size) for i in range(table_size+1,n+1): fac[i]=fac[i-1]*i%mod finv[n]=inv(fac[n]) for i in range(n-1,table_size,-1): finv[i]=finv[i+1]*(i+1)%mod table_size=n def binom(n,k): if n<0 or k<0: return 0 if k>n: return 0 if n>table_size: rebuild(n+10**4) return (fac[n]*finv[k]%mod)*finv[n-k]%mod def fpow(x,k): res=1 while k: if k&1: res=res*x%mod x=x*x%mod k>>=1 return res def inv(a): if a<table_size: return fac[a-1]*finv[a]%mod return fpow(a,mod-2) n=int(input()) p=list(map(lambda x:int(x)-1,input().split())) idx=[0]*n for i in range(n): idx[p[i]]=i ans=0 def f(a,b): return binom(a+b,a) seg=segtree([0]*n,lambda x,y:x+y,0) for i in range(n): lu=seg.query(0,idx[i]) ld=idx[i]-lu ru=seg.query(idx[i]+1,n) rd=n-1-idx[i]-ru ans+=f(lu,rd)*f(ru,ld) ans%=mod seg[idx[i]]=1 print(ans)