結果

問題 No.1036 Make One With GCD 2
ユーザー lam6er
提出日時 2025-04-16 00:35:40
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 933 bytes
コンパイル時間 465 ms
コンパイル使用メモリ 82,536 KB
実行使用メモリ 172,368 KB
最終ジャッジ日時 2025-04-16 00:37:49
合計ジャッジ時間 17,557 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 = list(map(int, input[1:n+1]))
    
    ans = 0
    prev_gcd = []  # list of tuples (g, cnt)
    
    for a in a_list:
        current_dict = {}
        
        # Process previous GCDs
        for g, cnt in prev_gcd:
            new_g = math.gcd(g, a)
            if new_g in current_dict:
                current_dict[new_g] += cnt
            else:
                current_dict[new_g] = cnt
        
        # Add the current element as a single-element subarray
        if a in current_dict:
            current_dict[a] += 1
        else:
            current_dict[a] = 1
        
        # Update prev_gcd for next iteration
        prev_gcd = list(current_dict.items())
        
        # Add the count of GCD 1 to the answer
        ans += current_dict.get(1, 0)
    
    print(ans)

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