結果

問題 No.1036 Make One With GCD 2
ユーザー lam6er
提出日時 2025-04-16 15:29:42
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 846 bytes
コンパイル時間 388 ms
コンパイル使用メモリ 81,772 KB
実行使用メモリ 171,304 KB
最終ジャッジ日時 2025-04-16 15:33:04
合計ジャッジ時間 17,846 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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