結果

問題 No.1577 織姫と彦星2
ユーザー kept1994
提出日時 2022-03-19 15:57:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 261 ms / 2,000 ms
コード長 1,345 bytes
コンパイル時間 413 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 97,488 KB
最終ジャッジ日時 2024-10-04 17:40:40
合計ジャッジ時間 12,437 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3

from bz2 import compress


def main():
    import math
    from collections import deque
    INF = 10 ** 16
    def bfs(edges: "List[to]", start_node: int) -> list:
        q = deque()
        dist = [INF] * len(edges)
        q.append(start_node)
        dist[start_node] = 0
        while q:
            now = q.popleft()
            for next in edges[now]:
                if dist[next] != INF:
                    continue
                q.append(next)
                dist[next] = dist[now] + 1
        return dist
    N = int(input())
    st, ed = map(int, input().split())
    stones = [st] + list(map(int, input().split())) + [ed]
    compressed = {}
    compressed_to_row = []
    routes = set(stones)
    for index, val in enumerate(sorted(list(routes))):
        compressed[val] = index
        compressed_to_row.append(val)
    # print(compressed)
    G = [[] for _ in range(N + 2)]
    for i in range(N + 2):
        for b in range(30):
            reachable = stones[i] ^ 1 << b
            if reachable in routes:
                G[compressed[stones[i]]].append(compressed[reachable])
                G[compressed[reachable]].append(compressed[stones[i]])
    d = bfs(G, compressed[st])
    print(d[compressed[ed]] - 1 if d[compressed[ed]] != INF else -1)
    return

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