結果
問題 |
No.2046 Ans Mod? Mod Ans!
|
ユーザー |
![]() |
提出日時 | 2022-08-19 22:22:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,765 ms / 4,000 ms |
コード長 | 1,416 bytes |
コンパイル時間 | 403 ms |
コンパイル使用メモリ | 82,108 KB |
実行使用メモリ | 121,472 KB |
最終ジャッジ日時 | 2024-12-16 08:23:04 |
合計ジャッジ時間 | 15,646 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
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)