結果

問題 No.1036 Make One With GCD 2
コンテスト
ユーザー brthyyjp
提出日時 2020-06-08 04:55:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 619 ms / 2,000 ms
コード長 1,533 bytes
コンパイル時間 457 ms
コンパイル使用メモリ 82,520 KB
実行使用メモリ 182,320 KB
最終ジャッジ日時 2024-09-16 14:26:19
合計ジャッジ時間 17,247 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline

def main():

    n = int(input())
    A = list(map(int, input().split()))

    import math
    def segfunc(x, y):
        return math.gcd(x, y)

    def init(init_val):
        # set_val
        for i in range(n):
            seg[i+num-1] = init_val[i]
        # built
        for i in range(num-2, -1, -1):
            seg[i] = segfunc(seg[2*i+1], seg[2*i+2])

    def update(k, x):
        k += num - 1
        seg[k] = x
        while k:
            k = (k-1)//2
            seg[k] = segfunc(seg[2*k+1], seg[2*k+2])

    def query(p, q):
        if q <= p:
            return ide_ele
        p += num - 1
        q += num - 2
        res = ide_ele
        while q-p>1:
            if p&1 == 0:
                res = segfunc(res, seg[p])
            if q&1 == 1:
                res = segfunc(res, seg[q])
                q -= 1
            p = p//2
            q = (q-1)//2
        if p == q:
            res = segfunc(res, seg[p])
        else:
            res = segfunc(segfunc(res, seg[p]), seg[q])
        return res

    # identity element
    ide_ele = 0

    # num: n以上の最小の2のべき乗
    num = 2**(n-1).bit_length()
    seg = [ide_ele]*2*num

    init(A)

    q = 1
    ans = 0
    for p in range(n):
        g =  query(p, q)
        while q < n and g != 1:
            g = math.gcd(g, A[q])
            q += 1
        if g != 1:
            break
        if p == q:
            q += 1
        ans += n-q+1
    print(ans)

if __name__ == '__main__':
    main()
0