結果

問題 No.1036 Make One With GCD 2
ユーザー lam6er
提出日時 2025-04-15 21:06:01
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 846 bytes
コンパイル時間 198 ms
コンパイル使用メモリ 82,040 KB
実行使用メモリ 171,848 KB
最終ジャッジ日時 2025-04-15 21:11:54
合計ジャッジ時間 15,768 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 = {}
    
    for a in A:
        curr_gcds = {}
        # Process previous GCDs
        for g, cnt in prev_gcds.items():
            new_g = math.gcd(g, a)
            if new_g in curr_gcds:
                curr_gcds[new_g] += cnt
            else:
                curr_gcds[new_g] = cnt
        # Add the current element as a subarray of length 1
        if a in curr_gcds:
            curr_gcds[a] += 1
        else:
            curr_gcds[a] = 1
        # Update the result with the count of GCD 1
        result += curr_gcds.get(1, 0)
        # Update prev_gcds for the next iteration
        prev_gcds = curr_gcds
    
    print(result)

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