結果

問題 No.1036 Make One With GCD 2
ユーザー aaaaaaaaaa2230
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0