https://programmers.co.kr/learn/courses/30/lessons/60062
코딩테스트 연습 - 외벽 점검
레스토랑을 운영하고 있는 "스카피"는 레스토랑 내부가 너무 낡아 친구들과 함께 직접 리모델링 하기로 했습니다. 레스토랑이 있는 곳은 스노우타운으로 매우 추운 지역이어서 내부 공사를 하
programmers.co.kr
문제 설명
레스토랑을 운영하고 있는 스카피는 레스토랑 내부가 너무 낡아 친구들과 함께 직접 리모델링하기로 했습니다. 레스토랑이 있는 곳은 스노우타운으로 매우 추운 지역이어서 내부 공사를 하는 도중에 주기적으로 외벽의 상태를 점검해야 할 필요가 있습니다.
레스토랑의 구조는 완전히 동그란 모양이고 외벽의 총 둘레는 n미터이며, 외벽의 몇몇 지점은 추위가 심할 경우 손상될 수도 있는 취약한 지점들이 있습니다. 따라서 내부 공사 도중에도 외벽의 취약 지점들이 손상되지 않았는지, 주기적으로 친구들을 보내서 점검을 하기로 했습니다. 다만, 빠른 공사 진행을 위해 점검 시간을 1시간으로 제한했습니다. 친구들이 1시간 동안 이동할 수 있는 거리는 제각각이기 때문에, 최소한의 친구들을 투입해 취약 지점을 점검하고 나머지 친구들은 내부 공사를 돕도록 하려고 합니다. 편의 상 레스토랑의 정북 방향 지점을 0으로 나타내며, 취약 지점의 위치는 정북 방향 지점으로부터 시계 방향으로 떨어진 거리로 나타냅니다. 또, 친구들은 출발 지점부터 시계, 혹은 반시계 방향으로 외벽을 따라서만 이동합니다.
외벽의 길이 n, 취약 지점의 위치가 담긴 배열 weak, 각 친구가 1시간 동안 이동할 수 있는 거리가 담긴 배열 dist가 매개변수로 주어질 때, 취약 지점을 점검하기 위해 보내야 하는 친구 수의 최솟값을 return 하도록 solution 함수를 완성해주세요.
제한 사항
- n은 1 이상 200 이하인 자연수입니다.
- weak의 길이는 1 이상 15 이하입니다.
- 서로 다른 두 취약점의 위치가 같은 경우는 주어지지 않습니다.
- 취약 지점의 위치는 오름차순으로 정렬되어 주어집니다.
- weak의 원소는 0 이상 n - 1 이하인 정수입니다.
- dist의 길이는 1 이상 8 이하입니다.
- dist의 원소는 1 이상 100 이하인 자연수입니다.
- 친구들을 모두 투입해도 취약 지점을 전부 점검할 수 없는 경우에는 -1을 return 해주세요.
입출력 예
| n | weak | dist | result |
| 12 | [1, 5, 6, 10] | [1, 2, 3, 4] | 2 |
| 12 | [1, 3, 4, 9, 10] | [3, 5, 7] | 1 |
EX01 )
- 4m를 이동할 수 있는 친구는 10m 지점에서 출발해 시계방향으로 돌아 1m 위치에 있는 취약 지점에서 외벽 점검을 마칩니다.
- 2m를 이동할 수 있는 친구는 4.5m 지점에서 출발해 6.5m 지점에서 외벽 점검을 마칩니다.
그 외에 여러 방법들이 있지만, 두 명보다 적은 친구를 투입하는 방법은 없습니다. 따라서 친구를 최소 두 명 투입해야 합니다.
EX02 )
7m를 이동할 수 있는 친구가 4m 지점에서 출발해 반시계 방향으로 점검을 돌면 모든 취약 지점을 점검할 수 있습니다. 따라서 친구를 최소 한 명 투입하면 됩니다.
문제 풀이
카카오 코딩 테스트 문제 해설
weak와 dist의 최대 길이는 각각 15와 8로, 주어지는 데이터의 개수가 적으므로 완전 탐색으로 해결할 수 있다.
1. 원형인 weak를 직선 형태로 고려한다.
2. 첫 시작점을 한 칸씩 이동하며, 모든 weak의 순서를 확인한다.
3. 순서를 고려하여 모든 dist의 경우의 수를 확인한다.
4. 최종적으로 가장 적은 친구를 보낼 수 있도록 한다.
1. 원형을 직선 형태로
원형으로 주어진 weak를 이용하여 문제를 풀어주기에는 까다롭기 때문에, 배열을 2배로 늘려주어 직선 형태로 만들어주고 문제를 풀기 시작한다.
2배로 늘려줄 때, 단순히 같은 값을 넣어주지 않고 n 값을 더해준 값을 추가해주도록 한다.
친구의 이동 거리를 더해가며 최종 점검 가능한 값을 비교해야 하기 때문에, 같은 값을 넣어 늘려주지 않고 n값을 더해 늘려준다.
weak = [ 1, 5, 6, 10 ]이 weak = [ 1, 5, 6, 10, 13, 17, 18, 22 ] 가 되도록 해준다.
length = len(weak)
for i in range(length):
weak.append(weak[i] + n)
2. 모든 weak, dist의 경우의 수
weak에서의 첫 스타트 부분을 length( = len(weak) [원형 형태]) 를 이용하여 순서대로 정해주도록 한다.
[1, 5, 6, 10] → [5, 6, 10, 13] → [ 6, 10, 13, 17] 등의 순서로 모든 weak의 순서를 고려할 수 있도록 한다.
친구들의 순서에 따라 최대 점검할 수 있는 위치가 달라지기 때문에, 순서를 고려하는 순열을 이용해준다.
순열은 저번에 풀었던 [BOJ/Python] 15686_치킨 배달 에서 조합을 이용해주었듯이 itertools를 이용해준다.
from itertools import permutations
...
for start in range(length):
for friends in list(permutations(dist, len(dist))):
cnt = 1
position = weak[start] + friends[cnt - 1]cnt는 필요한 친구의 수를 의미하며, position은 weak에서 시작한 지점으로부터 친구의 이동을 더하여 최종적으로 점검할 수 있는 위치를 의미한다.
3. 취약 지점을 모두 점검했는지?
이제 첫 친구를 보내어 취약 지점이 모두 점검되었는지 확인해주어야 한다.
weak의 시작 지점에 따라 순서대로 비교하며, 최종적으로 위치한 곳(position)이 weak의 지점보다 작다면, 아직 모든 취약 지점을 점검하지 못했다는 것이다.
따라서, 친구를 한 명 더 출발시키고 최종 위치를 다시 계산해준다.
for index in range(start, start + length):
if position < weak[index]:
cnt += 1
if cnt > len(dist):
break
position = weak[index] + friends[cnt - 1]
4. 마무리
가장 적은 친구를 보낼 수 있도록 min 함수를 이용하여 가장 적은 값을 리턴하도록 한다.
최종 값이 dist의 길이보다 큰 경우, 모든 친구들을 보내어도 모든 취약 지점을 점검할 수 없다는 것을 의미하므로 -1을 리턴해주도록 한다.
✅[ CODE ]
from itertools import permutations
def solution(n, weak, dist):
answer = len(dist) + 1
length = len(weak)
for i in range(length):
weak.append(weak[i] + n)
for start in range(length):
for friends in list(permutations(dist, len(dist))):
cnt = 1
position = weak[start] + friends[cnt - 1]
for index in range(start, start + length):
if position < weak[index]:
cnt += 1
if cnt > len(dist):
break
position = weak[index] + friends[cnt - 1]
answer = min(answer, cnt)
if answer > len(dist):
return -1
return answer