이 문제는 곰곰히 생각해보다가 풀기 힘들어 이 곳의 코드 풀이를 참고해 풀었다. 직접 푼 것이 아니어서 글을 쓸까 고민하다가 코드 방식과 알지 못했던 파이썬 언어 등 배울점이 많아서 참고용으로 글을 쓰게 되었다.
✏️ 문제
유명한 제빵사 김원웅은 빵집을 운영하고 있다. 원웅이의 빵집은 글로벌 재정 위기를 피해가지 못했고, 결국 심각한 재정 위기에 빠졌다.
원웅이는 지출을 줄이고자 여기저기 지출을 살펴보던 중에, 가스비가 제일 크다는 것을 알게되었다. 따라서 원웅이는 근처 빵집의 가스관에 몰래 파이프를 설치해 훔쳐서 사용하기로 했다.
빵집이 있는 곳은 R*C 격자로 표현할 수 있다. 첫째 열은 근처 빵집의 가스관이고, 마지막 열은 원웅이의 빵집이다.
원웅이는 가스관과 빵집을 연결하는 파이프를 설치하려고 한다. 빵집과 가스관 사이에는 건물이 있을 수도 있다. 건물이 있는 경우에는 파이프를 놓을 수 없다.
가스관과 빵집을 연결하는 모든 파이프라인은 첫째 열에서 시작해야 하고, 마지막 열에서 끝나야 한다. 각 칸은 오른쪽, 오른쪽 위 대각선, 오른쪽 아래 대각선으로 연결할 수 있고, 각 칸의 중심끼리 연결하는 것이다.
원웅이는 가스를 되도록 많이 훔치려고 한다. 따라서, 가스관과 빵집을 연결하는 파이프라인을 여러 개 설치할 것이다. 이 경로는 겹칠 수 없고, 서로 접할 수도 없다. 즉, 각 칸을 지나는 파이프는 하나이어야 한다.
원웅이 빵집의 모습이 주어졌을 때, 원웅이가 설치할 수 있는 가스관과 빵집을 연결하는 파이프라인의 최대 개수를 구하는 프로그램을 작성하시오.
🎈 입력
첫째 줄에 R과 C가 주어진다. (1 ≤ R ≤ 10,000, 5 ≤ C ≤ 500)
다음 R개 줄에는 빵집 근처의 모습이 주어진다. '.'는 빈 칸이고, 'x'는 건물이다. 처음과 마지막 열은 항상 비어있다.
🎈 출력
첫째 줄에 원웅이가 놓을 수 있는 파이프라인의 최대 개수를 출력한다.
🎈 입출력 예
<입력>
5 5
.xx..
..x..
.....
...x.
...x.
<출력>
2
<입력>
6 10
..x.......
.....x....
.x....x...
...x...xx.
..........
....x.....
<출력>
5
👩💻 내 코드
이 문제는 그리디 알고리즘, 깊이 우선 탐색을 이용해서 푸는 문제이다. 파이프라인을 최대로 설치하기 위한 방법은 아래와 같다.
방향을 우상향, 직진, 좌하향으로 나누었을 때, 가장 벽쪽에 설치할 수록 최대로 설치할 수 있다.
즉, 우상향 또는 좌하향에 설치할 수록 최대로 설치 가능하다.
위의 아이디어를 이용해서 문제를 풀면 아래와 같다. 문법의 자세한 설명은 주석으로 적어놓았다.
R, C = map(int, input().split())
maps = []
for _ in range(R):
maps.append(list(input()))
dirs=[(-1, 1), (0, 1), (1, 1)] # 우상향, 직진, 좌하향
end_line = C-1
def pipeline(y:int, x: int) -> bool: # 이 함수가 반환하는 자료형은 bool
# 현 위치에 파이프 생성
maps[y][x] = 'x'
# 만약 원웅이 빵집에 도달했다면 True 리턴
if x == end_line:
return True
for dy, dx in dirs:
ny, nx = y+dy, x+dx
# 다음 위치에 파이프 생성 가능한지 확인
# ny와 nx가 맵 안에 위치하고, '.'일 때 True 리턴
if 0 <= ny < R and 0 <= nx < C and maps[ny][nx] == '.':
# 끝까지 설치가 가능하다면
if pipeline(ny, nx):
return True
# 전부 조건이 안되면 False 리턴
return False
result = 0
for y in range(R):
if pipeline(y, 0):
# 파이프 라인 하나 설치 시
result += 1
print(result)