[Programmers]Python_외벽 점검

문제

문제 설명

레스토랑을 운영하고 있는 스카피는 레스토랑 내부가 너무 낡아 친구들과 함께 직접 리모델링 하기로 했습니다. 레스토랑이 있는 곳은 스노우타운으로 매우 추운 지역이어서 내부 공사를 하는 도중에 주기적으로 외벽의 상태를 점검해야 할 필요가 있습니다.

레스토랑의 구조는 완전히 동그란 모양이고 외벽의 총 둘레는 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 해주세요.

코드

def solution(n, weak, dist):
    answer = 0
    #오름차순 정렬
    dist.sort(reverse = True)
    #투입된 사람 수에 따른 수리할 수 있는 경우들을 저장
    repairs = [()]
    
    #가장 이동할 수 있는 거리가 긴 사람부터 투입!
    for cur_man in dist:
        #현재 투입된 사람이 수리할 수 있는 지점들을 저장
        cur_repairs = []
        answer += 1
        
        #취약한 지점마다 시작점을 지정하여 순회
        for idx, value in enumerate(weak):
            start = value
            #시작점 전에 있던 지점들도 한바퀴 더 돌았을때 도달하므로 n만큼 더해줘서 추가해준다
            cur_weak = weak[idx:] + [n + w for w in weak[:idx]]
            #현재 투입된 사람이 움직일 수 있는 사정거리 안에 있는 모든 취약 지점을 추가한다
            cur_repair = [weak % n for weak in cur_weak if (weak - start) <= cur_man]
            #다른 사람이 수리 가능한 곳과 합쳐볼 것이기 때문에 set()으로 추가한다
            cur_repairs.append(set(cur_repair))
            
        next_repairs = set()
        for i in cur_repairs: #현재 투입된 사람의 수리 가능 지점들
            for j in repairs: #이전에 투입된 사람들이 수리 가능한 지점들
                sum_ = i | set(j)
                #모두 수리 가능하다면 지금까지 투입된 사람 수 return
            	if len(sum_) == len(weak):
                    return answer
                #수리 가능한 지점들이 같은 경우가 있으므로 set에 tuple형식으로 추가해준다
                next_repairs.add(tuple(sum_))
            #아직 모든 지점을 수리할 수 없다면 repairs에 복사해주고 다름 사람 투입!
            repairs = next_repairs
    return -1

 

진행 과정

예제 #1

n = 12

weak = [1, 5, 6, 10]

dist = [1, 2, 3, 4]

 

Programmers

문제 설명에 이런 문장이 있다... "친구들은 출발 지점부터 시계, 혹은 반시계 방향으로 외벽을 따라서만 이동합니다."...

이 문장을 보고 "음... 시계 방향... 반시계 방향... 두 가지 경우를 따로 구현해야겠군..."이라고 생각했을 수도 있다.

하지만 0 -> 10으로 이동하든 10 -> 0으로 이동하든 결국에 0, 11, 10을 방문한 것은 똑같다. 따라서 한 방향으로 움직였을 경우만 생각해도된다.

 

1) dist = [4, 3, 2, 1] #오름차순 정렬

 

2) cur_man = 4 #현재 투입된 사람이 이동할 수 있는 거리

3) start = 1

4) cur_weak = [1, 5, 6, 10] # 수리해야할 지점

   cur_repair = [1, 5] # 수리 가능한 지점

5) strart = 5

6) cur_weak = [5, 6, 10, 13] # 수리해야할 지점

   cur_repair = [5, 6] # 수리 가능한 지점

7) strart = 6

8) cur_weak = [6, 10, 13, 17] # 수리해야할 지점

   cur_repair = [6, 10] # 수리 가능한 지점

9) strart = 10

10) cur_weak = [10, 13, 17, 18] # 수리해야할 지점

   cur_repair = [10, 1] # 수리 가능한 지점

11) cur_repairs = [{1, 5}, {5, 6}, {10, 6}, {1, 10}] #현재 투입된 사람이 수리 가능한 지점들

 

repairs = {(10, 6), (5, 6), (1, 10), (1, 5)} #현재까지의 수리 가능한 지점들

 

12) cur_man = 3

13) start = 1

14) cur_weak = [1, 5, 6, 10]

   cur_repair = [1]

15) strart = 5

16) cur_weak = [5, 6, 10, 13]

   cur_repair = [5, 6]

17) strart = 6

18) cur_weak = [6, 10, 13, 17]

   cur_repair = [6]

19) strart = 10

20) cur_weak = [10, 13, 17, 18]

   cur_repair = [10, 1]

21) cur_repairs = [{1}, {5, 6}, {6}, {1, 10}]

 

repairs = {(1, 10, 6), (1, 5, 6), (1, 10), (1, 5), (10, 5, 6), (5, 6), (1, 10, 5, 6)}

 

2명이 투입된 시점에서 모든 취약 지점을 수리할 수 있는 경우가 나왔다

 

22) return 2