문제

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

스타트링크의 사무실은 1×1크기의 정사각형으로 나누어져 있는 N×M 크기의 직사각형으로 나타낼 수 있다. 사무실에는 총 K개의 CCTV가 설치되어져 있는데, CCTV는 5가지 종류가 있다. 각 CCTV가 감시할 수 있는 방법은 다음과 같다.

1번 CCTV는 한 쪽 방향만 감시할 수 있다. 2번과 3번은 두 방향을 감시할 수 있는데, 2번은 감시하는 방향이 서로 반대방향이어야 하고, 3번은 직각 방향이어야 한다. 4번은 세 방향, 5번은 네 방향을 감시할 수 있다.

CCTV는 감시할 수 있는 방향에 있는 칸 전체를 감시할 수 있다. 사무실에는 벽이 있는데, CCTV는 벽을 통과할 수 없다. CCTV가 감시할 수 없는 영역은 사각지대라고 한다.

CCTV는 회전시킬 수 있는데, 회전은 항상 90도 방향으로 해야 하며, 감시하려고 하는 방향이 가로 또는 세로 방향이어야 한다.

지도에서 0은 빈 칸, 6은 벽, 1~5는 CCTV의 번호이다. 위의 예시에서 1번의 방향에 따라 감시할 수 있는 영역을 '#'로 나타내면 아래와 같다.

CCTV는 벽을 통과할 수 없기 때문에, 1번이 → 방향을 감시하고 있을 때는 6의 오른쪽에 있는 칸을 감시할 수 없다.

위의 예시에서 감시할 수 있는 방향을 알아보면 아래와 같다.

CCTV는 CCTV를 통과할 수 있다. 아래 예시를 보자.

위와 같은 경우에 2의 방향이 ↕ 3의 방향이 ←와 ↓인 경우 감시받는 영역은 다음과 같다.

사무실의 크기와 상태, 그리고 CCTV의 정보가 주어졌을 때, CCTV의 방향을 적절히 정해서, 사각 지대의 최소 크기를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 사무실의 세로 크기 N과 가로 크기 M이 주어진다. (1 ≤ N, M ≤ 8)

둘째 줄부터 N개의 줄에는 사무실 각 칸의 정보가 주어진다. 0은 빈 칸, 6은 벽, 1~5는 CCTV를 나타내고, 문제에서 설명한 CCTV의 종류이다.

CCTV의 최대 개수는 8개를 넘지 않는다.

출력

첫째 줄에 사각 지대의 최소 크기를 출력한다.

아이디어

✅ clone 함수 아이디어를 참고했다.

• 각 cctv가 볼 수 있는 방향의 수를 directions에 담아둔다.
• dfs를 이용해서 구할 수 있는 모든 수를 구한다.
• find함수를 이용해서 cctv가 볼 수 있는 영역을 '#'으로 표시한다.

✔️ 코드 설명
1. 입력을 받고 cctv의 위치를 따로 받아서 cctv리스트에 저장한다.
2. dfs에 배열과 현재 cctv 몇 개를 사용했는지를 표시하는 cnt를 함께 가지고 간다.
3. cctv가 주어진 수와 같으면 0인 영역의 수를 센다.
4. 아니면 해당 cctv의 번호를 가져와서 cctv방향에 있는 영역을 '#'으로 변경해준다.
5. 하나의 cctv의 방향을 설정했으면 다음 cctv의 방향도 정해야 하므로 다시 dfs함수를 호출한다.
6. cctv의 개수만큼 모두 방향을 정했으면 return을 하고 tmp는 다시 그 전의 상태의 배열로 돌려놓는다.

[#Error] 내 코드_python

혼자 풀려고 하다가 못 풀었다. 아직 dfs를 잘 구현하지 못하는 것 같다.

n, m = map(int, input().split())
arr = [list(map(int, input().split())) for x in range(n)]

d = [[-1, 0], [1, 0], [-1, 0], [0, -1]] #상하좌우
direction = [[],
             [[0], [1], [2], [3]],
             [[0, 2], [1, 3]],
             [[0, 1], [1, 2], [2, 3], [3, 0]],
             [[1, 2, 3], [0, 2, 3], [0, 1, 3], [0, 1, 2]],
             [[0, 1, 2, 3]]]

def find(x, y, dx, dy):
    tmp = []
    while 0 <= x+dx < n and 0 <= y+dy <m:
        x += dx
        y += dy
        if arr[x][y] == 6:
            break
        tmp.append([x, y])
    return tmp


cctv = []
w = []
for i in range(n):
    for j in range(m):
        if arr[i][j] in range(1, 6):
            cctv.append([arr[i][j], i, j])
        elif arr[i][j] == 6:
            w.append([i, j])

[#Clone] 함수_pypy3

clone함수 참고해서 다시 혼자 해봤는데 dfs는 아직 어려운 것 같다.

• python으로 하면 시간초과 된다. pypy3로 제출해야 통과된다.

import copy

n, m = map(int, input().split())
arr = [list(map(int, input().split())) for x in range(n)]
d = [[-1, 0], [0, -1], [1, 0], [0, 1]]  #상, 좌, 하, 우
directions = [[],
              [[0], [1], [2], [3]],
              [[0, 2], [1, 3]],
              [[0, 1], [1, 2], [2, 3], [3, 0]],
              [[1, 2, 3], [0, 2, 3], [0, 1, 3], [0, 1, 2]],
              [[0, 1, 2, 3]]
              ]


def find(x, y, direction, arrc):
    for k in direction:
        nx = x
        ny = y
        while 0 <= nx+d[k][0] < n and 0 <= ny+d[k][1] < m:
            nx += d[k][0]
            ny += d[k][1]
            if arrc[nx][ny] != 6:
                if arrc[nx][ny] == 0:
                    arrc[nx][ny] = '#'
            else:
                break


def dfs(a, cnt):
    global ans
    tmp = copy.deepcopy(a)
    if cnt == len(cctv):
        c = 0
        for i in range(len(tmp)):
            c += tmp[i].count(0)
        ans = min(ans, c)
        return
    cc, x, y = cctv[cnt]
    for i in range(len(directions[cc])):
        find(x, y, directions[cc][i], tmp)
        dfs(tmp, cnt+1)
        tmp = copy.deepcopy(a)


ans = 9999
cctv = []
for i in range(n):
    for j in range(m):
        if arr[i][j] in range(1, 6):
            cctv.append([arr[i][j], i, j])
dfs(arr, 0)
print(ans)

clone 함수 출처