結果

問題 No.2146 2 Pows
ユーザー lam6er
提出日時 2025-04-15 23:00:17
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,557 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 81,640 KB
実行使用メモリ 147,792 KB
最終ジャッジ日時 2025-04-15 23:02:08
合計ジャッジ時間 41,149 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 2 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from collections import deque

def main():
    N, A, B, C = map(int, sys.stdin.readline().split())
    max_z = 20  # Based on 2^20 being larger than 2e5
    INF = float('inf')

    # Precompute BFS for each z
    bfs_dist = []
    for z in range(max_z + 1):
        exponents = [pow(2, i, N) for i in range(z)]
        dist = [INF] * N
        dist[0] = 0
        q = deque()
        q.append(0)
        while q:
            s = q.popleft()
            for e in exponents:
                new_s = (s + e) % N
                if dist[new_s] > dist[s] + 1:
                    dist[new_s] = dist[s] + 1
                    q.append(new_s)
        bfs_dist.append(dist)
    
    ans = [INF] * N
    for k in range(N):
        for z in range(max_z + 1):
            s0 = pow(2, z, N)
            target = (k - s0) % N
            if target == 0:
                cost = (A + B) * 1 + C * z
                if cost < ans[k]:
                    ans[k] = cost
            else:
                dist = bfs_dist[z]
                s = dist[target]
                if s != INF:
                    cost = (A + B) * (s + 1) + C * z
                    if cost < ans[k]:
                        ans[k] = cost
        # If no solution found (ans[k] remains INF), but problem states S is non-empty
        # So there must be a solution. So handle INF if necessary.
        # But in practice, for the given constraints, there should be a solution.
    
    for k in range(N):
        print(ans[k] if ans[k] != INF else -1)

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