結果

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

ソースコード

diff #

import math
import sys

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:
        curr_gcds = {}
        # Process previous GCDs
        for g, cnt in prev_gcds:
            new_g = math.gcd(g, num)
            if new_g in curr_gcds:
                curr_gcds[new_g] += cnt
            else:
                curr_gcds[new_g] = cnt
        # Add the current number as a subarray of length 1
        if num in curr_gcds:
            curr_gcds[num] += 1
        else:
            curr_gcds[num] = 1
        # Update total with the count of GCD 1
        total += curr_gcds.get(1, 0)
        # Prepare prev_gcds for next iteration
        prev_gcds = list(curr_gcds.items())
    
    print(total)

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