結果
問題 | No.2046 Ans Mod? Mod Ans! |
ユーザー | とりゐ |
提出日時 | 2022-08-19 22:22:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,691 ms / 4,000 ms |
コード長 | 1,416 bytes |
コンパイル時間 | 332 ms |
コンパイル使用メモリ | 87,332 KB |
実行使用メモリ | 112,388 KB |
最終ジャッジ日時 | 2023-08-22 11:43:54 |
合計ジャッジ時間 | 15,712 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 168 ms
88,268 KB |
testcase_01 | AC | 201 ms
88,252 KB |
testcase_02 | AC | 290 ms
88,248 KB |
testcase_03 | AC | 812 ms
88,116 KB |
testcase_04 | AC | 620 ms
88,424 KB |
testcase_05 | AC | 788 ms
88,196 KB |
testcase_06 | AC | 632 ms
88,172 KB |
testcase_07 | AC | 683 ms
87,976 KB |
testcase_08 | AC | 220 ms
88,524 KB |
testcase_09 | AC | 221 ms
88,552 KB |
testcase_10 | AC | 226 ms
88,644 KB |
testcase_11 | AC | 245 ms
88,712 KB |
testcase_12 | AC | 240 ms
88,608 KB |
testcase_13 | AC | 827 ms
100,072 KB |
testcase_14 | AC | 693 ms
96,748 KB |
testcase_15 | AC | 1,020 ms
105,596 KB |
testcase_16 | AC | 1,061 ms
105,624 KB |
testcase_17 | AC | 976 ms
103,464 KB |
testcase_18 | AC | 1,653 ms
103,476 KB |
testcase_19 | AC | 951 ms
103,708 KB |
testcase_20 | AC | 1,691 ms
112,388 KB |
ソースコード
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 debug(self,s=10): print([self.get(i) for i in range(min(self.n,s))]) n=int(input()) a=sorted(list(map(int,input().split()))) ans=0 for i in range(n): ans+=a[i]*(n-1-i) m=2*10**5+10 seg=segtree([0]*m,lambda x,y:x+y,0) seg_cnt=segtree([0]*m,lambda x,y:x+y,0) for i in range(n-1,-1,-1): ai=a[i] tmp=1 while tmp*ai<m: cnt=seg_cnt.query(ai*tmp,min(m,ai*(tmp+1))) sm=seg.query(ai*tmp,min(m,ai*(tmp+1))) ans-=sm-cnt*ai*tmp tmp+=1 seg_cnt.update(ai,1) seg.update(ai,ai) print(ans)