結果
問題 |
No.3 ビットすごろく
|
ユーザー |
![]() |
提出日時 | 2020-03-13 01:20:32 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 54 ms / 5,000 ms |
コード長 | 1,033 bytes |
コンパイル時間 | 111 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 12,544 KB |
最終ジャッジ日時 | 2024-07-01 09:42:02 |
合計ジャッジ時間 | 2,392 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 33 |
ソースコード
#!/usr/bin/env python3 import sys input = sys.stdin.readline import heapq INF = 10**9 n = int(input()) edge = [[] for _ in range(n)] for i in range(1, n+1): step = bin(i).count("1") if i - step > 0: edge[i-1].append(i-step-1) if i + step <= n: edge[i-1].append(i+step-1) class Dijkstra: def __init__(self, adj): self.adj = adj self.dist = [INF] * len(adj) self.q = [] def calc(self, start): self.dist[start] = 0 heapq.heappush(self.q, (0, start)) while len(self.q) != 0: prov_cost, src = heapq.heappop(self.q) if self.dist[src] < prov_cost: continue for dest in self.adj[src]: if self.dist[dest] > self.dist[src] + 1: self.dist[dest] = self.dist[src] + 1 heapq.heappush(self.q, (self.dist[dest], dest)) return self.dist DIJK = Dijkstra(edge) DIJK.calc(0) if DIJK.dist[-1] == INF: print(-1) else: print(DIJK.dist[-1] + 1)