結果
| 問題 |
No.1036 Make One With GCD 2
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-03-27 23:35:34 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 722 ms / 2,000 ms |
| コード長 | 1,190 bytes |
| コンパイル時間 | 861 ms |
| コンパイル使用メモリ | 82,560 KB |
| 実行使用メモリ | 270,880 KB |
| 最終ジャッジ日時 | 2024-11-06 16:43:15 |
| 合計ジャッジ時間 | 19,241 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 41 |
ソースコード
from math import gcd
class SWAG:
def __init__(self,f):
self.st1 = []
self.st2 = []
self.rst1 = []
self.rst2 = []
self.f = f
def calc_all(self):
calc = 0
if self.st1:
calc = self.f(calc,self.rst1[-1])
if self.st2:
calc = self.f(calc,self.rst2[-1])
return calc
def pop_front(self):
if self.st1 == []:
self.st1 = self.st2[::-1]
self.rst1 = self.st1[:]
for i in range(1,len(self.st1)):
self.rst1[i] = self.f(self.rst1[i-1],self.st1[i])
self.st2 = []
self.rst2 = []
self.st1.pop()
self.rst1.pop()
def push_back(self,val):
if self.st2 == []:
self.rst2.append(val)
else:
self.rst2.append(self.f(self.rst2[-1],val))
self.st2.append(val)
n = int(input())
A = list(map(int,input().split()))
swag = SWAG(gcd)
ans = 0
now = 0
for i,a in enumerate(A):
while now < n and swag.calc_all() != 1:
swag.push_back(A[now])
now += 1
if swag.calc_all() == 1:
ans += n-now+1
swag.pop_front()
print(ans)