結果
問題 | No.1300 Sum of Inversions |
ユーザー | persimmon-persimmon |
提出日時 | 2021-02-15 13:19:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,201 ms / 2,000 ms |
コード長 | 1,151 bytes |
コンパイル時間 | 272 ms |
コンパイル使用メモリ | 86,964 KB |
実行使用メモリ | 155,960 KB |
最終ジャッジ日時 | 2023-09-30 06:43:24 |
合計ジャッジ時間 | 32,985 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 74 ms
71,292 KB |
testcase_01 | AC | 75 ms
71,184 KB |
testcase_02 | AC | 77 ms
70,972 KB |
testcase_03 | AC | 944 ms
120,456 KB |
testcase_04 | AC | 935 ms
120,076 KB |
testcase_05 | AC | 783 ms
108,504 KB |
testcase_06 | AC | 1,095 ms
145,516 KB |
testcase_07 | AC | 1,037 ms
125,716 KB |
testcase_08 | AC | 1,128 ms
148,688 KB |
testcase_09 | AC | 1,107 ms
148,516 KB |
testcase_10 | AC | 656 ms
122,272 KB |
testcase_11 | AC | 655 ms
122,300 KB |
testcase_12 | AC | 930 ms
120,484 KB |
testcase_13 | AC | 893 ms
119,668 KB |
testcase_14 | AC | 1,201 ms
155,960 KB |
testcase_15 | AC | 1,093 ms
148,256 KB |
testcase_16 | AC | 963 ms
121,240 KB |
testcase_17 | AC | 622 ms
118,424 KB |
testcase_18 | AC | 703 ms
110,924 KB |
testcase_19 | AC | 813 ms
116,128 KB |
testcase_20 | AC | 853 ms
116,184 KB |
testcase_21 | AC | 834 ms
116,180 KB |
testcase_22 | AC | 755 ms
107,704 KB |
testcase_23 | AC | 1,069 ms
145,408 KB |
testcase_24 | AC | 792 ms
110,764 KB |
testcase_25 | AC | 699 ms
123,256 KB |
testcase_26 | AC | 682 ms
122,980 KB |
testcase_27 | AC | 734 ms
109,748 KB |
testcase_28 | AC | 1,133 ms
154,840 KB |
testcase_29 | AC | 838 ms
116,348 KB |
testcase_30 | AC | 1,105 ms
148,912 KB |
testcase_31 | AC | 759 ms
108,008 KB |
testcase_32 | AC | 788 ms
110,652 KB |
testcase_33 | AC | 461 ms
124,928 KB |
testcase_34 | AC | 468 ms
123,524 KB |
testcase_35 | AC | 543 ms
134,476 KB |
testcase_36 | AC | 698 ms
155,260 KB |
ソースコード
# 0-indexed binary indexed tree class BIT: def __init__(self, n): self.n = n self.data = [0]*(n+1) self.el = [0]*(n+1) # sum of [0,i) sum(a[:i]) def sum(self, i): if i==0:return 0 s = 0 while i > 0: s += self.data[i] i -= i & -i return s def add(self, i, x): i+=1 self.el[i] += x while i <= self.n: self.data[i] += x i += i & -i # sum of [l,r) sum(a[l:r]) def sumlr(self, i, j): return self.sum(j) - self.sum(i) # a[i] def get(self,i): i+=1 return self.el[i] def main1(n,a): mod=998244353 aa=a[:] aa.sort() d={x:i for i,x in enumerate(aa)} bit1n=BIT(n) bit1s=BIT(n) bit2n=BIT(n) bit2s=BIT(n) bit3n=BIT(n) bit3s=BIT(n) for x in a: idx=d[x] bit1n.add(idx,1) bit1s.add(idx,x) n2=bit1n.sumlr(idx+1,n)%mod bit2n.add(idx,n2) bit2s.add(idx,bit1s.sumlr(idx+1,n)%mod+x*n2) n3=bit2n.sumlr(idx+1,n)%mod bit3n.add(idx,n3) bit3s.add(idx,bit2s.sumlr(idx+1,n)%mod+x*n3) return bit3s.sum(n)%mod if __name__=='__main__': n=int(input()) a=list(map(int,input().split())) ret1=main1(n,a) print(ret1)