結果

問題 No.3 ビットすごろく
ユーザー FromBooska
提出日時 2023-09-12 20:06:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 119 ms / 5,000 ms
コード長 1,043 bytes
コンパイル時間 135 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 79,200 KB
最終ジャッジ日時 2024-06-30 07:46:54
合計ジャッジ時間 4,036 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

# うーん、愚直にダイクストラで間に合うのか

N = int(input())
edges = [[] for i in range(N+1)]
for i in range(1, N+1):
    num = bin(i).count('1')
    if i-num >= 1:
        edges[i].append((i-num, 1))
    if i+num <= N:
        edges[i].append((i+num, 1))

from heapq import heappush, heappop
INF = 10 ** 18
def dijkstra(s, n, connect): #(始点, ノード数)
    distance = [INF] * n
    que = [(0, s)] #(distance, node)
    distance[s] = 0
    confirmed = [False] * n # ノードが確定済みかどうか
    while que:
        w,v = heappop(que)
        if distance[v]<w:
            continue
        confirmed[v] = True
        for to, cost in connect[v]: # ノード v に隣接しているノードに対して
            if confirmed[to] == False and distance[v] + cost < distance[to]:
                distance[to] = distance[v] + cost
                heappush(que, (distance[to], to))
    return distance
 
distance = dijkstra(1, N+1, edges)
ans = distance[N]+1
if ans >= INF:
    print(-1)
else:
    print(ans)
0