結果

問題 No.1036 Make One With GCD 2
コンテスト
ユーザー tamato
提出日時 2020-04-25 15:38:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,110 ms / 2,000 ms
コード長 1,789 bytes
コンパイル時間 282 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 255,944 KB
最終ジャッジ日時 2024-09-16 13:43:05
合計ジャッジ時間 25,003 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


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

    # min
    def STfunc(a, b):
        if a < b:
            a, b = b, a
        if b == 0:
            return a
        while a % b:
            a, b = b, a%b
        return 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.N * (self.K + 1))
            for i, a in enumerate(A):
                self.table[i] = 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 * self.N] = STfunc(self.table[i + (k - 1) * self.N],
                                                            self.table[j + (k - 1) * self.N])

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

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

    ST = SparseTable(A)
    ans = 0
    ok_prev = N
    for i in range(N-1, -1, -1):
        if A[i] == 1:
            ans += N - i
            ok_prev = i
            continue
        if ST.query(i, N) != 1:
            continue
        ok = ok_prev
        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
        ok_prev = ok
    print(ans)


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