結果
問題 | No.1036 Make One With GCD 2 |
ユーザー |
![]() |
提出日時 | 2020-04-25 00:06:28 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,848 bytes |
コンパイル時間 | 337 ms |
コンパイル使用メモリ | 82,264 KB |
実行使用メモリ | 260,444 KB |
最終ジャッジ日時 | 2024-11-07 03:18:14 |
合計ジャッジ時間 | 9,015 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 TLE * 10 |
ソースコード
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)])import sysinput = sys.stdin.readlinen = int(input())a = list(map(int, input().split()))sp = SparseTable(a)r = 0ans = 0for l in range(0, n):r = max(l + 1, r)while True:if r == n + 1:breakif sp.query(l, r) != 1:r += 1else:breakans += n - r + 1print(ans)