문제
[원문 링크]
유명한 제빵사 김원웅은 빵집을 운영하고 있다. 원웅이의 빵집은 글로벌 재정 위기를 피해가지 못했고, 결국 심각한 재정 위기에 빠졌다. 원웅이는 지출을 줄이고자 여기저기 지출을 살펴보던 중에, 가스비가 제일 크다는 것을 알게되었다. 따라서 원웅이는 근처 빵집의 가스관에 몰래 파이프를 설치해 훔쳐서 사용하기로 했다. 빵집이 있는 곳은 R*C 격자로 표현할 수 있다. 첫째 열은 근처 빵집의 가스관이고, 마지막 열은 원웅이의 빵집이다. 원웅이는 가스관과 빵집을 연결하는 파이프를 설치하려고 한다. 빵집과 가스관 사이에는 건물이 있을 수도 있다. 건물이 있는 경우에는 파이프를 놓을 수 없다. 가스관과 빵집을 연결하는 모든 파이프라인은 첫째 열에서 시작해야 하고, 마지막 열에서 끝나야 한다. 각 칸은 오른쪽, 오른쪽 위 대각선, 오른쪽 아래 대각선으로 연결할 수 있고, 각 칸의 중심끼리 연결하는 것이다. 원웅이는 가스를 되도록 많이 훔치려고 한다. 따라서, 가스관과 빵집을 연결하는 파이프라인을 여러 개 설치할 것이다. 이 경로는 겹칠 수 없고, 서로 접할 수도 없다. 즉, 각 칸을 지나는 파이프는 하나이어야 한다. 원웅이 빵집의 모습이 주어졌을 때, 원웅이가 설치할 수 있는 가스관과 빵집을 연결하는 파이프라인의 최대 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 R과 C가 주어진다. (1 ≤ R ≤ 10,000, 5 ≤ C ≤ 500) 다음 R개 줄에는 빵집 근처의 모습이 주어진다. ‘.’는 빈 칸이고, ‘x’는 건물이다. 처음과 마지막 열은 항상 비어있다.
출력
첫째 줄에 원웅이가 놓을 수 있는 파이프라인의 최대 개수를 출력한다.
해설
이 문제는 가장 왼쪽 열에서 시작해 가장 오른쪽 열로 도달하는 가장 윗 길을, 각 행마다 겹치지 않게 찾아주면 되는 간단한 문제다.
- 각 행(
y)을 for문을 돌리고, 열(x)은 0부터 큐에 넣어주고,1 2
for i in range(N): que.append([i, 0])
- DFS를 돌면서
x+1과(y+1, y, y-1)칸들을 탐색하면서1 2 3 4 5 6
while que: y, x = que.pop() vis[y][x] = 1 nx = x + 1 for j in range(1, -2, -1): ny = y + j
- 장애물이 없으며, 동시에 방문하지 않은 칸을 큐에 추가하고,
1 2
if 0 <= ny < N and brd[ny][nx] == '.' and not vis[ny][nx]: que.append((ny, nx))
x가 가장 오른쪽 열(M-1)에 도달한 경우, DFS를 멈추고 정답(cnt)에 1을 더해준다.1 2 3 4
if x == M-1: cnt += 1 que.clear() break
유의할 점으로는, 파이프로 쓰이지 않았더라도, 방문했던 칸은 다시 방문하지 않도록 해주었는데, 이는 어차피 해당 칸은 결국 막혀 있어 쓰이지 않을 것이기 때문이다.
예시
밑의 예시 그림을 보면서 큐에 어떤 좌표가 있는지 따라가면 이해하기 쉬울 것이다.
큐에 (0, 0)을 넣고 시작한다.
que = [(0, 0)]
(0, 0) (1번칸) 탐색시, 위를 제외한 앞, 밑칸으로 갈 수 있다.
que = [(1, 1), (0, 1)]
(0, 1) (2번칸) 탐색시, 위, 앞, 밑 모두 막혀 있다.
que = [(1, 1)]
(1, 1) (3번칸) 탐색시, 밑칸으로 갈 수 있다.
que = [(2, 2)]
(2, 2) (4번칸) 탐색시, 위, 앞칸으로 갈 수 있다.
que = [(3, 2), (3, 3)]
(3, 3) (5번칸) 탐색시, 위, 앞, 밑 모두 갈 수 있다.
que = [(3, 2), (4, 3), (4, 2), (4, 1)]
(4, 1) (6번칸) 탐색시, x가 M-1인 4에 도달 했으므로, DFS 종료.
cnt에 1을 더해주고, 방문 기록은 그대로 남겨두고, 다음 열에서 DFS 다시 시작, 마지막 열까지 모두 진행하면 끝이다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import sys
input = sys.stdin.readline
def main():
N, M = map(int, input().split())
brd = [input().strip() for _ in range(N)]
vis = [[0] * M for _ in range(N)]
cnt = 0
que = []
for i in range(N):
que.append([i, 0])
while que:
y, x = que.pop()
vis[y][x] = 1
if x == M-1:
cnt += 1
que.clear()
break
nx = x + 1
for j in range(1, -2, -1):
ny = y + j
if 0 <= ny < N and brd[ny][nx] == '.' and not vis[ny][nx]:
que.append((ny, nx))
return cnt
if __name__ == "__main__":
print(main())
결과
