(Python/파이썬) - 백준(BOJ) 3109번 : 빵집

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

3109번: 빵집

---

• 문제 : 

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

 

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

 

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

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

 

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

 

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

 

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

 

 

 

---

• 풀이 :

DFS를 그리디하게 탐색하는 문제였습니다.

 

알고리즘의 흐름입니다.

 

1. 0부터 R-1번째 행의 0번째 열을 출발지점으로 삼고 dfs를 돌립니다. 이 때, dfs의 인자는 x와 y의 좌표입니다.

 

2. dfs함수에선 종료조건으로 y좌표가 마지막 열인 C-1번까지 갔는지를 체크합니다.

 

3. y좌표는 재귀 호출마다 무조건 1씩 증가합니다, 그러나 x좌표는 [-1, 0, 1]의 세 갈래로 변화할 수 있습니다. 이 때 가장 많은 파이프라인을 설치하기 위해선 오름차순으로 탐색해야 합니다. 따라서 for 문에서 dx는 -1, 0, 1 순으로 변화합니다.

 

4. ax와 ay의 값이 board를 벗어나지 않으면서 아직 방문하지 않은 좌표라면 해당 좌표에 대해 dfs를 호출합니다. 이 때, 중요한 점은 dfs의 return값이 false인 경우 (마지막 열 까지 갈 수 없고 길이 막힌경우) 가 아니라면 파이프를 설치할 수 있다는 것이므로 다음 경로는 탐색하지 않아도 됩니다. 따라서 return True를 해줍니다.

 

ex ) dx = -1 일 때 경로가 있다면 dx가 0일때는 탐색하지 않아도 됨.

 

5. 최종적으로 dfs가 True를 반환한다면 answer를 1 증가시킵니다.

 

 

 

---

• 소스코드 : 

import sys
input = sys.stdin.readline
answer = 0
def dfs(x, y):
    if y == C-1:
        return True
    for dx in [-1,0,1]:
        ax = x + dx
        ay = y + 1
        if 0 <= ax < R and 0 <= ay < C:
            if board[ax][ay] != "x" and visited[ax][ay] == -1:
                visited[ax][ay] = 1
                if dfs(ax, ay):
                    return True
    return False


R, C = map(int,input().split())
visited = [[-1 for _ in range(C)] for _ in range(R)]
board = [list(input().strip()) for _ in range(R)]
for i in range(R):
    if dfs(i, 0): answer += 1
print(answer)