結果
問題 | No.1036 Make One With GCD 2 |
ユーザー |
![]() |
提出日時 | 2020-04-24 23:51:50 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,919 bytes |
コンパイル時間 | 673 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 269,280 KB |
最終ジャッジ日時 | 2024-11-07 03:14:52 |
合計ジャッジ時間 | 46,599 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 29 TLE * 12 |
ソースコード
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()))sp = SparseTable(a)l, r = 0, 0ans = 0while True:while True:if r == n + 1:breakif l == r:r += 1continueif sp.query(l, r) == 1:ans += n - r + 1breakelse:r += 1l += 1if l > r:r = lif l == n + 1:breakprint(ans)