結果
問題 | No.1300 Sum of Inversions |
ユーザー | shakayami |
提出日時 | 2020-11-27 21:52:49 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,065 ms / 2,000 ms |
コード長 | 1,565 bytes |
コンパイル時間 | 472 ms |
コンパイル使用メモリ | 86,820 KB |
実行使用メモリ | 107,820 KB |
最終ジャッジ日時 | 2023-10-01 05:28:33 |
合計ジャッジ時間 | 29,051 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 92 ms
71,768 KB |
testcase_01 | AC | 92 ms
71,272 KB |
testcase_02 | AC | 93 ms
71,556 KB |
testcase_03 | AC | 886 ms
101,880 KB |
testcase_04 | AC | 839 ms
105,396 KB |
testcase_05 | AC | 718 ms
100,740 KB |
testcase_06 | AC | 948 ms
104,640 KB |
testcase_07 | AC | 907 ms
102,976 KB |
testcase_08 | AC | 1,018 ms
106,032 KB |
testcase_09 | AC | 988 ms
105,604 KB |
testcase_10 | AC | 597 ms
97,476 KB |
testcase_11 | AC | 599 ms
96,704 KB |
testcase_12 | AC | 852 ms
100,428 KB |
testcase_13 | AC | 821 ms
104,152 KB |
testcase_14 | AC | 1,065 ms
107,820 KB |
testcase_15 | AC | 999 ms
105,284 KB |
testcase_16 | AC | 893 ms
103,188 KB |
testcase_17 | AC | 594 ms
95,520 KB |
testcase_18 | AC | 668 ms
98,980 KB |
testcase_19 | AC | 784 ms
101,864 KB |
testcase_20 | AC | 780 ms
102,476 KB |
testcase_21 | AC | 787 ms
102,920 KB |
testcase_22 | AC | 719 ms
100,196 KB |
testcase_23 | AC | 1,031 ms
104,396 KB |
testcase_24 | AC | 733 ms
102,268 KB |
testcase_25 | AC | 633 ms
97,660 KB |
testcase_26 | AC | 622 ms
97,676 KB |
testcase_27 | AC | 674 ms
100,144 KB |
testcase_28 | AC | 1,040 ms
106,628 KB |
testcase_29 | AC | 752 ms
102,752 KB |
testcase_30 | AC | 1,007 ms
105,344 KB |
testcase_31 | AC | 706 ms
100,668 KB |
testcase_32 | AC | 744 ms
101,584 KB |
testcase_33 | AC | 427 ms
103,360 KB |
testcase_34 | AC | 440 ms
107,508 KB |
testcase_35 | AC | 427 ms
105,188 KB |
testcase_36 | AC | 489 ms
107,184 KB |
ソースコード
from collections import defaultdict class fenwick_tree(): n=1 data=[0 for i in range(n)] def __init__(self,N): self.n=N self.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+=1 while(p<=self.n): self.data[p-1]+=x p+=p& -p def 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=0 while(r>0): s+=self.data[r-1] r-=r&-r return s N=int(input()) A=[int(i) for i in input().split()] B=sorted([(A[i],i) for i in range(N)]) I=[B[i][1] for i in range(N)] GL=fenwick_tree(N) GR=fenwick_tree(N) SL=fenwick_tree(N) SR=fenwick_tree(N) GL0=[0 for i in range(N)] GR0=[0 for i in range(N)] SL0=[0 for i in range(N)] SR0=[0 for i in range(N)] for i in range(N-1,-1,-1): GL0[I[i]]=GL.sum(0,I[i]) SL0[I[i]]=SL.sum(0,I[i]) GL.add(I[i],1) SL.add(I[i],A[I[i]]) for i in range(N): GR0[I[i]]=GR.sum(I[i],N) SR0[I[i]]=SR.sum(I[i],N) GR.add(I[i],1) SR.add(I[i],A[I[i]]) L=defaultdict(int) R=defaultdict(int) ''' for i in range(N-1,-1,-1): GL0[i]-=L[A[i]] SL0[i]-=A[i]*L[A[i]] L[A[i]]+=1 for i in range(N): GR0[i]-=R[A[i]] SR0[i]-=A[i]*R[A[i]] R[A[i]]+=1 print(GL0) print(GR0) print(SL0) print(SR0) ''' ans=0 mod=998244353 for i in range(N): ans+=GL0[i]*SR0[i]+SL0[i]*GR0[i]+A[i]*GL0[i]*GR0[i] ans%=mod print(ans)