結果
問題 |
No.1036 Make One With GCD 2
|
ユーザー |
![]() |
提出日時 | 2020-04-24 22:10:25 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 894 ms / 2,000 ms |
コード長 | 1,752 bytes |
コンパイル時間 | 370 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 171,504 KB |
最終ジャッジ日時 | 2024-09-16 13:18:04 |
合計ジャッジ時間 | 17,280 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 41 |
ソースコード
""" sliding window aggrigation モノイドのqueueを管理する fold_all: すべてをたたみこむ popleft append """ from itertools import accumulate from collections import deque class SWAG: def __init__(self, operator_M, e_M): self.op_M = operator_M self.op_rev = lambda x,y: self.op_M(y,x) self.e_M = e_M self.q = deque([]) self.accL = [] self.accR = e_M self.L = self.R = 0 def build(self,lst): self.q = deque(lst) self.L = len(lst) self.accL = list(accumulate(reversed(lst),self.op_rev)) def __len__(self): return self.L + self.R def fold_all(self): if self.L: return self.op_M(self.accL[-1],self.accR) else: return self.accR def append(self,x): self.q.append(x) self.accR = self.op_M(self.accR,x) self.R += 1 def popleft(self): if self.L: self.accL.pop() self.L -= 1 return self.q.popleft() elif self.R: v = self.q.popleft() self.L,self.R = self.R-1,0 self.accL = list(accumulate(reversed(self.q),self.op_rev)) self.accR = self.e_M return v else: assert 0 def __repr__(self): return "{}\naccL:{}, accR:{}".format(self.q,self.accL,self.accR) # coding: utf-8 # Your code here! import sys readline = sys.stdin.readline read = sys.stdin.read n,*a = [int(i) for i in read().split()] from math import gcd q = SWAG(gcd,0) ans = r = 0 for l,al in enumerate(a): if len(q): q.popleft() else: r = l while r<n and gcd(q.fold_all(),a[r]) != 1: q.append(a[r]) r += 1 ans += n-r print(ans)