#540548 (Python3) No.697 池の数はいくつか

• 問題: No.697 池の数はいくつか
• ユーザー: 双六
• 提出日時: 2020-08-26 15:16:40
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: TLE
• 実行時間: -
• コード長: 1,123 bytes
• コンパイル時間: 263 ms
• コンパイル使用メモリ: 12,800 KB
• 実行使用メモリ: 305,408 KB
• 最終ジャッジ日時: 2024-11-25 16:24:32
• 合計ジャッジ時間: 43,100 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	29 ms; 163,712 KB
testcase_01	AC	29 ms; 10,880 KB
testcase_02	AC	30 ms; 10,880 KB
testcase_03	AC	29 ms; 10,880 KB
testcase_04	AC	27 ms; 10,880 KB
testcase_05	AC	28 ms; 10,880 KB
testcase_06	AC	30 ms; 10,752 KB
testcase_07	AC	29 ms; 10,880 KB
testcase_08	AC	28 ms; 10,752 KB
testcase_09	AC	29 ms; 10,752 KB
testcase_10	AC	29 ms; 11,008 KB
testcase_11	AC	29 ms; 10,880 KB
testcase_12	AC	28 ms; 10,880 KB
testcase_13	AC	29 ms; 10,752 KB
testcase_14	AC	29 ms; 10,752 KB
testcase_15	AC	28 ms; 10,752 KB
testcase_16	AC	30 ms; 10,880 KB
testcase_17	AC	29 ms; 10,624 KB
testcase_18	AC	29 ms; 10,752 KB
testcase_19	AC	29 ms; 10,880 KB
testcase_20	AC	30 ms; 10,752 KB
testcase_21	AC	28 ms; 10,752 KB
testcase_22	AC	29 ms; 10,880 KB
testcase_23	AC	30 ms; 10,880 KB
testcase_24	AC	647 ms; 26,880 KB
testcase_25	AC	674 ms; 26,880 KB
testcase_26	AC	664 ms; 27,008 KB
testcase_27	AC	664 ms; 26,880 KB
testcase_28	AC	660 ms; 26,880 KB
testcase_29	TLE	-
testcase_30	AC	5,780 ms; 152,576 KB
testcase_31	TLE	-
testcase_32	AC	5,662 ms; 152,448 KB
testcase_33	AC	5,655 ms; 152,576 KB
testcase_34	MLE	-

ソースコード

# import sys; input = sys.stdin.buffer.readline
# sys.setrecursionlimit(10**7)
# from collections import defaultdict
from collections import deque

def getlist():
	return list(map(int, input().split()))

def BFS(H, W, L, visit, start):
	h, w = start
	Q = deque(); Q.append([h, w])
	while Q:
		y, x = Q.popleft()
		if L[y - 1][x] == 1 and visit[y - 1][x] == 0:
			visit[y - 1][x] = 1
			Q.append([y - 1, x])
		if L[y + 1][x] == 1 and visit[y + 1][x] == 0:
			visit[y + 1][x] = 1
			Q.append([y + 1, x])
		if L[y][x - 1] == 1 and visit[y][x - 1] == 0:
			visit[y][x - 1] = 1
			Q.append([y, x - 1])
		if L[y][x + 1] == 1 and visit[y][x + 1] == 0:
			visit[y][x + 1] = 1
			Q.append([y, x + 1])

#処理内容
def main():
	H, W = getlist()
	L = []
	L.append([0] * (W + 2))
	for i in range(H):
		l = getlist()
		L.append([0] + l + [0])
	L.append([0] * (W + 2))

	visit = [[0] * (W + 2) for i in range(H + 2)]
	# print(L)
	cnt = 0
	for i in range(1, H + 1):
		for j in range(1, W + 1):
			if visit[i][j] == 0 and L[i][j] == 1:
				cnt += 1
				BFS(H + 2, W + 2, L, visit, [i, j])

	print(cnt)

if __name__ == '__main__':
	main()