문제 설명
레스토랑을 운영하고 있는 "스카피"는 레스토랑 내부가 너무 낡아 친구들과 함께 직접 리모델링 하기로 했습니다. 레스토랑이 있는 곳은 스노우타운으로 매우 추운 지역이어서 내부 공사를 하는 도중에 주기적으로 외벽의 상태를 점검해야 할 필요가 있습니다.
레스토랑의 구조는 완전히 동그란 모양이고 외벽의 총 둘레는 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 해주세요.
문제 풀이
먼저 처음에 원형 구조에 시계, 반 시계 구조인 경우는 배열을 두 배 늘려, 선형 구조를 취하는 것이 쉽게 해결할 수 있다는 것을 보았다. 그래서 두 배로 늘린 후, 다소 간단한 알고리즘을 수행했는데 시간 복잡도에서 걸렸다. 알고리즘은 이렇다.
-
weak 배열을 2배로 늘려줌.
-
dist 조합을 구해준다.
-
weak 시작점을 하나씩 옮겨가며 모든 weak ([weak + len(weak]) 갈 수 있다면 return.
3-1. 갈 수 없다면, 친구를 한명 더 투입
3-2. 이전 친구가 갔던 weak 다음 weak 점 부터 시작함.
이렇게 나눌 수 있겠다. 코드를 보자면,
from itertools import permutations # 친구들의 조합을 위함.
import copy # 확장해주기 위함.
def solution(n, weak, dist):
answer = 0
append_list = copy.deepcopy(weak)
friend = []
# 먼저 확장된 weak리스트 작성.
for i in range(len(weak)):
append_list.append(n + weak[i])
#weak 시작점을 하나 씩 옮겨 주려함.
for i , start in enumerate(weak):
# 친구 조합별로 살펴봄.( 아마 이거 때문에 시간 초과 나는듯 )
for friend_comb in permutations(dist):
cnt = 1 # 친구 1명 동원
pos = start # 처음 시작점으로 위치 잡아줌
for f in friend_comb:
pos += f
# 친구가 갈 수 있는 거리만큼 더해줌.
if pos < append_list[i + len(weak) -1] :
# 만약 그 위치가 시작점에서 모든 weak 점을 못가면
cnt += 1
# 친구 1명더 동원.
pos = [w for w in append_list[i+1: i + len(weak)] if w > pos][0]
# 이전 친구가 갔던 다음 weak점 부터 시작.
else:
# 만약 그 위치가 시작점에서 모든 weak 점을 갈 수 있다면
friend.append(cnt)
break
answer = min(friend)
if len(friend) == 0:
return -1
return answer결과는 테스트 케이스 몇 가지에서 런타임 에러가 발생하게 된다...🥲
아마 모든 조합을 살펴봐서 그런듯 하다...그럼 아마 모든 조합 보다 많이 갈 수 있는 친구를 우선적으로 하면 되지 않을까 한다. 그런데 찾아보니 사람들도 다 permutations를 사용하는데 통과가 돼서 저 코드를 고치는 방향으로 돌렸다. (절대 귀찮아서가 아님 ㅎ )
놀랍게도 마지막에 저 부분에서 friend 배열이 비워져 있는 상태에서 min 함수를 사용해서 에러가 나는 것이었다..
answer = min(friend)
if len(friend) == 0:
return -1
return answer여기에서 이렇게 바꿔주면 결과는 통과!
if len(friend) == 0:
return -1
return min(friend)