結果
問題 | No.1036 Make One With GCD 2 |
ユーザー |
|
提出日時 | 2021-10-02 12:38:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 754 ms / 2,000 ms |
コード長 | 2,105 bytes |
コンパイル時間 | 267 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 177,888 KB |
最終ジャッジ日時 | 2024-07-20 06:17:03 |
合計ジャッジ時間 | 20,612 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 41 |
ソースコード
import sysfrom math import gcdinput = sys.stdin.buffer.readlineclass SegTree:"""https://qiita.com/takayg1/items/c811bd07c21923d7ec69 から拝借しています。init(init_val, ide_ele): 配列init_valで初期化 O(N)update(k, x): k番目の値をxに更新 O(logN)query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)"""def __init__(self, init_val, segfunc, ide_ele):"""init_val: 配列の初期値segfunc: 区間にしたい操作ide_ele: 単位元n: 要素数num: n以上の最小の2のべき乗tree: セグメント木(1-index)"""n = len(init_val)self.segfunc = segfuncself.ide_ele = ide_eleself.num = 1 << (n - 1).bit_length()self.tree = [ide_ele] * 2 * self.num# 配列の値を葉にセットfor i in range(n):self.tree[self.num + i] = init_val[i]# 構築していくfor i in range(self.num - 1, 0, -1):self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1])def update(self, k, x):"""k番目の値をxに更新k: index(0-index)x: update value"""k += self.numself.tree[k] = xwhile k > 1:self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1])k >>= 1def query(self, l, r):"""[l, r)のsegfuncしたものを得るl: index(0-index)r: index(0-index)"""res = self.ide_elel += self.numr += self.numwhile l < r:if l & 1:res = self.segfunc(res, self.tree[l])l += 1if r & 1:res = self.segfunc(res, self.tree[r - 1])l >>= 1r >>= 1return resN = int(input())A = tuple(map(int, input().split()))seg = SegTree(A, gcd, 0)l = 0ans = 0for r in range(1, N + 1):while seg.query(l, r) == 1 and r - l >= 1:l += 1ans += lprint(ans)