[BOJ] 백준 3109번 빵집 - 파이썬(Python)

문제

유명한 제빵사 김원웅은 빵집을 운영하고 있다. 원웅이의 빵집은 글로벌 재정 위기를 피해가지 못했고, 결국 심각한 재정 위기에 빠졌다.

원웅이는 지출을 줄이고자 여기저기 지출을 살펴보던 중에, 가스비가 제일 크다는 것을 알게되었다. 따라서 원웅이는 근처 빵집의 가스관에 몰래 파이프를 설치해 훔쳐서 사용하기로 했다.

빵집이 있는 곳은 R*C 격자로 표현할 수 있다. 첫째 열은 근처 빵집의 가스관이고, 마지막 열은 원웅이의 빵집이다.

원웅이는 가스관과 빵집을 연결하는 파이프를 설치하려고 한다. 빵집과 가스관 사이에는 건물이 있을 수도 있다. 건물이 있는 경우에는 파이프를 놓을 수 없다.

가스관과 빵집을 연결하는 모든 파이프라인은 첫째 열에서 시작해야 하고, 마지막 열에서 끝나야 한다. 각 칸은 오른쪽, 오른쪽 위 대각선, 오른쪽 아래 대각선으로 연결할 수 있고, 각 칸의 중심끼리 연결하는 것이다.

원웅이는 가스를 되도록 많이 훔치려고 한다. 따라서, 가스관과 빵집을 연결하는 파이프라인을 여러 개 설치할 것이다. 이 경로는 겹칠 수 없고, 서로 접할 수도 없다. 즉, 각 칸을 지나는 파이프는 하나이어야 한다.

원웅이 빵집의 모습이 주어졌을 때, 원웅이가 설치할 수 있는 가스관과 빵집을 연결하는 파이프라인의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 R과 C가 주어진다. (1 ≤ R ≤ 10,000, 5 ≤ C ≤ 500)

다음 R개 줄에는 빵집 근처의 모습이 주어진다. '.'는 빈 칸이고, 'x'는 건물이다. 처음과 마지막 열은 항상 비어있다.

출력

첫째 줄에 원웅이가 놓을 수 있는 파이프라인의 최대 개수를 출력한다.

 

https://www.acmicpc.net/problem/3109

3109번: 빵집

 

💡 풀이 및 코드

r, c = map(int, input().split())
data = []
for _ in range(r):
    data.append(list(input()))

def dfs(array, x, y, visited):
    global pipe
    if y == c-2:
        pipe = array
        return 
    for i in range(3):
        nx = x + dx[i]
        if 0 <= nx and nx < r and not visited[nx][y+1] and data[nx][y+1] == '.' :
            visited[nx][y+1] = 1
            dfs(array + [(nx, y+1)], nx, y+1, visited)
            if len(pipe) == c-2:
               break

visited = [[0]*c for _ in range(r)]
dx = [-1, 0, 1]
result = 0
for i in range(r):
    pipe = []
    if data[i][1] == '.':
        visited[i][1] = 1
        dfs([(i, 1)], i, 1, visited)
        if pipe:
            for x, y in pipe:
                data[x][y] = 'P'
            result += 1
print(result)

맨 처음 풀이한 코드이다. 각 열에서 출발하여 가장 첫번 째행과 최대한 가까운 길을 선택하는 그리디 문제이다. 길 탐색을 위해서는 DFS를 사용하였다. 

테스트케이스는 모두 맞추었지만 제출 시 시간초과가 떴다. 한 질문 글에서 해답을 찾을 수 있었다. visited 테이블을 만들어서 이전에 탐색한 길은 다시 탐색하지 않도록 해야했다.

r, c = map(int, input().split())
data = []
for _ in range(r):
    data.append(list(input()))

def dfs(array, x, y):
    if y == c-2:
        return True
    data[x][y] = 'P'
    for i in range(3):
        nx = x + dx[i]
        if 0 <= nx and nx < r and data[nx][y+1] == '.' :
            if dfs(array + [(nx, y+1)], nx, y+1):
                return True
    return False

dx = [-1, 0, 1]
result = 0
for i in range(r):
    if data[i][1] == '.':
        if dfs([(i, 1)], i, 1):
            result += 1
print(result)

다른 블로그 풀이 글을 보고 코드를 개선해보았다. 이전처럼 파이프를 놓을 수 있는 경로에 대해서만 data 값을 변경해주지 않고  탐색 시마다 무조건 data 값을 변경해준다면 길 탐색과 방문확인을 동시에 하는 효과를 줄 수 있었다. visited 테이블을 없애고 data에 직접 방문의 표시를 하면서, 파이프를 놓을 수 있는지 여부는 return 값을 통해 알 수 있도록 했다.