結果

問題 No.1036 Make One With GCD 2
ユーザー tamato
提出日時 2020-04-24 22:12:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,479 ms / 2,000 ms
コード長 1,612 bytes
コンパイル時間 225 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 241,456 KB
最終ジャッジ日時 2024-09-16 13:19:00
合計ジャッジ時間 35,757 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


def main():
    import sys
    from math import gcd
    input = sys.stdin.readline

    # min
    def STfunc(a, b):
        return gcd(a, b)

    # クエリは0-indexedで[l, r)
    class SparseTable():
        def __init__(self, A):
            # A: 処理したい数列
            self.N = len(A)
            self.K = self.N.bit_length() - 1
            self.table = [[0] * (self.K + 1) for _ in range(self.N)]
            for i, a in enumerate(A):
                self.table[i][0] = a
            for k in range(1, self.K + 1):
                for i in range(self.N):
                    j = i + (1 << (k - 1))
                    if j <= self.N - 1:
                        self.table[i][k] = STfunc(self.table[i][k - 1], self.table[j][k - 1])
                    # else:
                    #    self.table[i][k] = self.table[i][k-1]

        def query(self, l, r):
            # [l, r)の最小値を求める
            k = (r - l).bit_length() - 1
            return STfunc(self.table[l][k], self.table[r - (1 << k)][k])

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

    ST = SparseTable(A)
    ans = 0
    for i in range(N):
        if A[i] == 1:
            ans += N - i
            continue
        if ST.query(i, N) != 1:
            continue
        ok = N-1
        ng = i
        mid = (ok + ng) // 2
        while ok - ng > 1:
            if ST.query(i, mid+1) == 1:
                ok = mid
            else:
                ng = mid
            mid = (ok+ng) // 2
        ans += N - ok
    print(ans)


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