結果

問題 No.1036 Make One With GCD 2
ユーザー lam6er
提出日時 2025-04-16 00:36:43
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 879 bytes
コンパイル時間 831 ms
コンパイル使用メモリ 81,280 KB
実行使用メモリ 170,940 KB
最終ジャッジ日時 2025-04-16 00:41:22
合計ジャッジ時間 17,524 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 34 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from math import gcd

def main():
    input = sys.stdin.read().split()
    n = int(input[0])
    a = list(map(int, input[1:n+1]))
    
    total = 0
    prev_gcds = {}
    
    for num in a:
        current_gcds = {}
        # Process previous GCDs
        for g in prev_gcds:
            new_g = gcd(g, num)
            if new_g in current_gcds:
                current_gcds[new_g] += prev_gcds[g]
            else:
                current_gcds[new_g] = prev_gcds[g]
        # Add the current number as a subarray of length 1
        if num in current_gcds:
            current_gcds[num] += 1
        else:
            current_gcds[num] = 1
        # Update total with the count of GCD 1
        total += current_gcds.get(1, 0)
        # Update prev_gcds for next iteration
        prev_gcds = current_gcds
    
    print(total)

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