結果

問題 No.3 ビットすごろく
ユーザー pajannat
提出日時 2021-04-11 17:56:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 196 ms / 5,000 ms
コード長 975 bytes
コンパイル時間 116 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 14,976 KB
最終ジャッジ日時 2024-06-27 12:38:53
合計ジャッジ時間 4,696 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
	N = int(input())

	# 10進数を2進数に変換
	def BitTrans(n):
		i = 0
		bit = []
		while n > 0:
			bit.insert(0, n % 2)
			n = int(n / 2)
			i += 1
		return bit

	# 2進数表記に含まれる1の数を数える
	def BitCount(bit):
		cnt = 0
		for i in bit:
			cnt += i
		return cnt

	list = []
	cnt = 0
	for i in range(1,N):
		if (i - BitCount(BitTrans(i))) > 0:
			list.append([i,(i - BitCount(BitTrans(i)))])
			cnt += 1
		if (i + BitCount(BitTrans(i))) < N + 1:
			list.append([i, (i + BitCount(BitTrans(i)))])
			cnt += 1


	from collections import deque

	n, m = N, cnt

	graph = [[] for _ in range(n+1)]

	for j in range(m):
		a, b =  list[j][0], list[j][1]
		graph[a].append(b)

	dist = [-1] * (n+1)
	dist[0] = 0
	dist[1] = 1

	d = deque()
	d.append(1)

	while d:
		v = d.popleft()
		for i in graph[v]:
			if dist[i] != -1:
				continue
			dist[i] = dist[v] + 1
			d.append(i)

	ans = dist[1:]
	print(ans[-1])

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