結果
問題 | No.2327 Inversion Sum |
ユーザー |
![]() |
提出日時 | 2023-05-28 13:57:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 154 ms / 2,000 ms |
コード長 | 1,605 bytes |
コンパイル時間 | 339 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 86,988 KB |
最終ジャッジ日時 | 2024-12-26 22:24:44 |
合計ジャッジ時間 | 4,618 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
mod=998244353def cmb(n,r):if r<0 or r>n:return 0return ((g1[n]*g2[r]%mod)*g2[n-r])%modN=300000g1=[1]*(N+3)for i in range(2,N+3):g1[i]=g1[i-1]*i%modg2=[0]*len(g1)g2[-1]=pow(g1[-1],mod-2,mod)for i in range(N+1,-1,-1):g2[i]=g2[i+1]*(i+1)%modinv=[0]*(N+3)for i in range(1,N+3):inv[i]=g2[i]*g1[i-1]%modN,M=map(int,input().split())P=[-1]*Nfor i in range(M):p,k=map(int,input().split())P[k-1]=p-1class fenwick_tree():n=1data=[0 for i in range(n)]def __init__(self,N):self.n=Nself.data=[0 for i in range(N)]def add(self,p,x):assert 0<=p<self.n,"0<=p<n,p={0},n={1}".format(p,self.n)p+=1while(p<=self.n):self.data[p-1]+=xp+=p& -pdef sum(self,l,r):assert (0<=l and l<=r and r<=self.n),"0<=l<=r<=n,l={0},r={1},n={2}".format(l,r,self.n)return self.sum0(r)-self.sum0(l)def sum0(self,r):s=0while(r>0):s+=self.data[r-1]r-=r&-rreturn sBIT=fenwick_tree(N+2)V=g1[N-M]ANS=0for i in range(N):if P[i]>=0:ANS=(ANS+V*BIT.sum(P[i]+1,N))%modBIT.add(P[i],1)ANS=(ANS+((((N-M)*(N-M-1))//2)%mod)*V*inv[2])%modC=[1]*(N+1)C[0]=0for i in range(N):if P[i]>=0:C[P[i]+1]=0for i in range(N):C[i+1]+=C[i]D=0V=V*inv[C[-1]]%modfor i in range(N):if P[i]>=0:ANS=(ANS+(V*(C[-1]-C[P[i]+1])%mod)*D)%modelse:D+=1P=P[::-1]D=0for i in range(N):if P[i]>=0:ANS=(ANS+(V*C[P[i]]%mod)*D)%modelse:D+=1print(ANS)