[PYTHON] 백준 1388 - 바닥 장식

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

성능 요약

메모리: 116048 KB, 시간: 140 ms

분류

깊이 우선 탐색, 그래프 이론, 그래프 탐색, 구현

문제 설명

형택이는 건축가이다. 지금 막 형택이는 형택이의 남자 친구 기훈이의 집을 막 완성시켰다. 형택이는 기훈이 방의 바닥 장식을 디자인했고, 이제 몇 개의 나무 판자가 필요한지 궁금해졌다. 나무 판자는 크기 1의 너비를 가졌고, 양수의 길이를 가지고 있다. 기훈이 방은 직사각형 모양이고, 방 안에는 벽과 평행한 모양의 정사각형으로 나누어져 있다.

이제 ‘-’와 ‘|’로 이루어진 바닥 장식 모양이 주어진다. 만약 두 개의 ‘-’가 인접해 있고, 같은 행에 있다면, 두 개는 같은 나무 판자이고, 두 개의 ‘|’가 인접해 있고, 같은 열에 있다면, 두 개는 같은 나무 판자이다.

기훈이의 방 바닥을 장식하는데 필요한 나무 판자의 개수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 방 바닥의 세로 크기N과 가로 크기 M이 주어진다. 둘째 줄부터 N개의 줄에 M개의 문자가 주어진다. 이것은 바닥 장식 모양이고, '-‘와 ’|‘로만 이루어져 있다. N과 M은 50 이하인 자연수이다.

출력

첫째 줄에 문제의 정답을 출력한다.

---

아이디어, 문제풀이

• 바닥 장식은 방향이 다르다고 카운트가 증가하는게 아니다.
• 방향이다르고, 그 방향에 맞춰서 연결되어있을때 1개로 간주한다.
• count 배열을 만들어, 같은 방향, 같은모양인 경우 count = 0을 대입, 새어주지 않는 방법으로 구현했다.
• 방향에 맞춰 방향 for문을 나눠줬다.

---

TROUBLE SHOOTING

• 이 문제를 실전 테스트에서 풀었을땐, 아이디어에 적힌 방향에 맞춰 for문 나누기를 해본적이 없어 풀지 못했다. 그래서 억지로 재귀함수를 노트에 써가며 구현했다. 아래는 재귀함수로 구현한 코드!

  import sys
  input = sys.stdin.readline
  
  n, m = map(int, input().split())
  
  graph = []
  for _ in range(n):
      a = list(map(str, input().strip()))
      graph.append(a)
  
  visit = [[0] * m for _ in range(n)]
  
  def dfs(x, y):
      visit[x][y] = 1
      if graph[x][y] == '-':
          if y + 1 < m and graph[x][y + 1] == '-' and visit[x][y + 1] == 0:
              dfs(x, y + 1)
      elif graph[x][y] == '|':
          if x + 1 < n and graph[x + 1][y] == '|' and visit[x + 1][y] == 0:
              dfs(x + 1, y)
  
  count = 0
  for i in range(n):
      for j in range(m):
          if visit[i][j] == 0:
              count += 1
              dfs(i, j)
  
  print(count)

  정말 쉽게 구현한것 처럼 보이지만…. 이 재귀를 생각해내고 구현하는데에만 40분 넘게 쓴것 같다… 진짜 어려웠다.
  
• 코딩테스트가 끝나고, 실전에 적용할수 있게끔 stack을 활용해서 풀어보고자 했다. for문을 나누는 방법을 알아와서, 그 방법을 토대로 코드를 구현했다. 근데 계속 답이 틀리게 나오기도 했고, 직접 print를 찍어봤을때

  4 4
  ----
  ----
  ----
  ----

  이게 [[1,0,0,0],[1,1,1,1],[1,1,1,1],[1,1,1,1]] 이런 결과를 만들어 냈는데, 이 문제를 해결해낼 방법을 떠올리는 과정이 은근 재밌었다. 왜냐면 하나하나 다 바꿔봤기 때문에… 재밌는게 맞나?… 결론을 말하자면, 코드에서 문제라기 보단, 출력과정에서 단순히 dfs를 호출해선 안됐다. 모든 정점에서 시작해야지만, 모두를 체크할 수 있다. 출력값 부분을 바꿔줘야 했던 문제였는데, 한번도 이런식으로 함수를 for문안에 넣어 반복 실행해 본적이 없어서 새로웠다. 분명 함수 안에 어떤 연산을 집어넣어준다면 저렇게 출력하지 않아도 됐을텐데, 아직 그 방법을 떠올리기엔 내 머리는 부족하다.

  for i in range(n):
      for j in range(m):
          if visit[i][j] == 0:
              bfs(i, j)

  위와같이 작성하면, 모든 정점에 대해서 bfs를 시작하기 때문에 멈춤 없이 구현이 가능하고, 올바른 답이 나온다.
  

---

코드

#https://www.acmicpc.net/problem/1388
#바닥 장식
#1388

from collections import deque
import sys
input = sys.stdin.readline

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

graph = []
for _ in range(n):
    a = list(map(str, input().strip()))
    graph.append(a)

visit = [[0] * m for _ in range(n)]
count = [[1] * m for _ in range(n)]

dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]

def bfs(a, b):
    que = deque()
    que.append((a,b))
    visit[a][b] = 1

    while que:
        x, y = que.popleft()

        for i in range(2):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < n and 0 <= ny < m and visit[nx][ny] == 0:
                if graph[x][y] == '|':
                    if graph[nx][ny] == '|':
                        visit[nx][ny] = 1
                        count[nx][ny] = 0
                        que.append((nx, ny))

        for i in range(2, 4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < n and 0 <= ny < m and visit[nx][ny] == 0:
                if graph[x][y] == '-':
                    if graph[nx][ny] == '-':
                        visit[nx][ny] = 1
                        count[nx][ny] = 0
                        que.append((nx, ny))

    return

for i in range(n):
    for j in range(m):
        if visit[i][j] == 0:
            bfs(i, j)

print(sum(map(sum, count)))