結果
問題 |
No.3 ビットすごろく
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
# うーん、愚直にダイクストラで間に合うのか 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)