[코드트리] 완전탐색 - 원 모양으로 되어있는 방

코드트리 네이버 커리큘럼 - 완전탐색

• 원 모양으로 되어있는 방 (Python)
  https://www.codetree.ai/cote/14/problems/a-room-in-a-circle/description

Code

N = int(input())
room = []
for _ in range(N):
    room.append(int(input()))

ans = []

for i in range(N):
    move = 1
    move_list = []

    for j in range(i, N + i - 1):
        next = room[(j + 1) % N]
        move_list.append(next * move)
        move += 1

    ans.append(sum(move_list))

print(min(ans))

풀이 및 해설

• 시작점은 거리 0, 그 뒤로는 거리가 1부터 ~
• 시작점을 인덱스 0부터 N-1까지 하나씩 뽑아서 → 그 옆방부터 방에 들어갈 수 있는 사람 수 * 거리 를 move_list에 담아줌
• 현재 move_list에 담긴 총 합을 ans에 담아줌
• 최종 정답은 ans의 최솟값