結果

問題 No.1036 Make One With GCD 2
ユーザー Shinya FujitaShinya Fujita
提出日時 2024-11-02 00:02:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,089 bytes
コンパイル時間 294 ms
コンパイル使用メモリ 82,584 KB
実行使用メモリ 243,944 KB
最終ジャッジ日時 2024-11-02 00:02:15
合計ジャッジ時間 9,576 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SegmentTree:
    def __init__(self, op, e, v):
        self.e = e
        self.op = op
        self.n = len(v)
        self.N = 2 ** (self.n-1).bit_length()
        self.seg_data = [e for _ in range(self.N-1)] \
            + v + [e for _ in range(self.N-self.n)]
        for i in range(2*self.N-2, 0, -2):
            self.seg_data[(i-1)//2] = op(self.seg_data[i], self.seg_data[i-1])
    
    def __len__(self):
        return self.n
    
    def __getitem__(self, i):
        return self.seg_data[self.N-1+i]
    
    def __setitem__(self, i, x):
        idx = self.N - 1 + i
        self.seg_data[idx] = x
        while idx:
            idx = (idx-1) // 2
            self.seg_data[idx] = self.op(self.seg_data[2*idx+1], self.seg_data[2*idx+2])
    
    def query(self, i, j):
        # [i, j)
        if i == j:
            return None
        else:
            idx1 = self.N - 1 + i
            idx2 = self.N - 2 + j # 閉区間にする
            result = self.e
            while idx1 < idx2 + 1:
                if idx1&1 == 0: # idx1が偶数
                    result = self.op(result, self.seg_data[idx1])
                if idx2&1 == 1: # idx2が奇数
                    result = self.op(result, self.seg_data[idx2])
                    idx2 -= 1
                idx1 //= 2
                idx2 = (idx2 - 1)//2
            return result
    
    def find_left(self, i):
        if self.query(0, i+1) != 1:
            return -1
        
        if self.seg_data[self.N-1+i] == 1:
            return i
        
        left = 0; right = i
        while left + 1 < right:
            mid = (left + right) // 2
            if self.query(mid, i+1) != 1:
                right = mid
            else:
                left = mid
        return left


from math import gcd

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

def op(x, y):
    if x is None:
        return y
    if y is None:
        return x
    return gcd(x, y)


seg = SegmentTree(op=op, e=None, v=A)
ans = N * (N+1) // 2
for i in range(N):
    left = seg.find_left(i)
    ans -= i - left

print(ans)
0