結果

問題 No.1946 ロッカーの問題
ユーザー gew1fw
提出日時 2025-06-12 20:47:55
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 752 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,104 KB
実行使用メモリ 120,112 KB
最終ジャッジ日時 2025-06-12 20:49:30
合計ジャッジ時間 2,417 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 5 WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    N = int(input[ptr])
    ptr += 1
    M = int(input[ptr])
    ptr += 1
    A = list(map(int, input[ptr:ptr + M])) if M > 0 else []
    
    in_A = [False] * (N + 1)
    for a in A:
        in_A[a] = True
    
    sum_prev = [0] * (N + 1)
    x = [0] * (N + 1)
    
    for k in range(1, N + 1):
        if in_A[k]:
            xk = (sum_prev[k] + 1) % 2
        else:
            xk = sum_prev[k] % 2
        x[k] = xk
        
        m = 2 * k
        while m <= N:
            sum_prev[m] += xk
            if sum_prev[m] >= 2:
                sum_prev[m] -= 2
            m += k
    
    total = sum(x[1:N+1])
    print(N - total)

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