結果

問題 No.1036 Make One With GCD 2
ユーザー gew1fw
提出日時 2025-06-12 15:38:46
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 953 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 82,760 KB
実行使用メモリ 163,544 KB
最終ジャッジ日時 2025-06-12 15:39:15
合計ジャッジ時間 18,184 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 34 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
import math

def main():
    input = sys.stdin.read().split()
    N = int(input[0])
    A = list(map(int, input[1:N+1]))
    
    result = 0
    prev_gcds = dict()
    
    for i in range(N):
        num = A[i]
        if num == 1:
            count = i + 1  # 子数组的数量是i+1个
            result += count
            current_gcds = {1: count}
        else:
            current_gcds = dict()
            current_gcds[num] = 1
            for g in prev_gcds:
                new_gcd = math.gcd(g, num)
                if new_gcd in current_gcds:
                    current_gcds[new_gcd] += prev_gcds[g]
                else:
                    current_gcds[new_gcd] = prev_gcds[g]
            # 检查是否有GCD为1的情况
            if 1 in current_gcds:
                result += current_gcds[1]
        # 更新prev_gcds
        prev_gcds = current_gcds.copy()
    
    print(result)

if __name__ == "__main__":
    main()
0