結果

問題 No.3 ビットすごろく
ユーザー nagitaosu
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/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)
0