結果
問題 | No.1036 Make One With GCD 2 |
ユーザー |
![]() |
提出日時 | 2020-04-24 22:04:20 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,814 bytes |
コンパイル時間 | 621 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 278,464 KB |
最終ジャッジ日時 | 2024-11-07 02:25:30 |
合計ジャッジ時間 | 47,315 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 24 TLE * 14 -- * 3 |
ソースコード
def gcd(a: int, b: int) -> int:"""a, bの最大公約数(greatest common divisor: GCD)を求める計算量: O(log(min(a, b)))"""if b == 0:return areturn gcd(b, a%b)class SparseTable:"""区間取得クエリをO(1)で答えるデータ構造をO(NlogN)で構築するquery(l, r): 区間[l, r)に対するクエリに答える"""def __init__(self, array):n = len(array)self.row_size = n.bit_length()# log_tableを構築する# log_table = [0, 0, 1, 1, 2, 2, 2, 2, ...]self.log_table = [0] * (n + 1)for i in range(2, n + 1):self.log_table[i] = self.log_table[i//2] + 1# sparse_tableを構築するself.sparse_table = [[0] * n for _ in range(self.row_size)]for i in range(n):self.sparse_table[0][i] = array[i]for row in range(1, self.row_size):for i in range(n - (1 << row) + 1):self.sparse_table[row][i] = self._merge(self.sparse_table[row - 1][i], \self.sparse_table[row - 1][i + (1 << row - 1)])def _merge(self, num1, num2):"""冪等律と結合律を満たす演算を行う"""return gcd(num1, num2)def query(self, l, r):"""区間[l, r)に対するクエリに答える"""row = self.log_table[r - l]return self._merge(self.sparse_table[row][l], self.sparse_table[row][r - (1 << row)])n = int(input())a = list(map(int, input().split())) + [1]n += 1sp = SparseTable(a)ans = 0for i in range(n):ok = nng = iwhile abs(ok - ng) > 1:mid = (ok + ng) // 2if sp.query(i, mid) == 1:ok = midelse:ng = midans += (n - ok)print(ans)