음료수 얼려먹기

이 문제는 이것이 코딩테스트다 책 149페이지의 문제이다.

DFS의 제일 기초적인 문제이다. 구현자체는 빠르게 했는데 변수이름을 잘못적어 오류를 찾느라 시간이 생각보다 오래 걸렸다...

DFS문제는 주기적으로 풀어주어야겠다.

 

 

 

아래는 나의 코드이다.

import sys
myMap=[]
move=[(-1,0),(1,0),(0,1),(0,-1)]

N,M= map(int,sys.stdin.readline().split())

visited=[[True]*N for _ in range(M)]

for i in range(M):
    temp = (sys.stdin.readline().split())
    myMap.append(list(map(int,temp[0])))


print(myMap)

def search(start_x,start_y):
    if start_x >= M or start_y >= N or start_x < 0 or start_y < 0:  #맵을 벗어나는지 체크
        return 0
    if visited[start_x][start_y]==True and myMap[start_x][start_y]==0: #방문,빈틀 체크
        visited[start_x][start_y]=False
        for i in move:
            next_x = start_x+i[0]
            next_y = start_y+i[1]
            search(next_x,next_y)
        return 1    	
    else:
        return 0


count=0
for i in range(M):
    for j in range(N):
        count +=  search(i,j) #최초실행함수에서만 결국 1을 리턴받으니 리턴값의 합은 아이스크림의 수
print(count)

 

 

아래는 예시코드

n, m = map(int, input().split())

# 2차원 리스트의 맵 정보 입력 받기
graph = []
for i in range(n):
    graph.append(list(map(int, input())))

# DFS로 특정한 노드를 방문한 뒤에 연결된 모든 노드들도 방문
def dfs(x, y):
    # 주어진 범위를 벗어나는 경우에는 즉시 종료
    if x <= -1 or x >= n or y <= -1 or y >= m:
        return False
    # 현재 노드를 아직 방문하지 않았다면
    if graph[x][y] == 0:
        # 해당 노드 방문 처리
        graph[x][y] = 1
        # 상, 하, 좌, 우의 위치들도 모두 재귀적으로 호출
        dfs(x - 1, y)
        dfs(x, y - 1)
        dfs(x + 1, y)
        dfs(x, y + 1)
        return True
    return False

# 모든 노드(위치)에 대하여 음료수 채우기
result = 0
for i in range(n):
    for j in range(m):
        # 현재 위치에서 DFS 수행
        if dfs(i, j) == True:
            result += 1

print(result) # 정답 출력