153. 외벽 점검

프로그래머스

1. Python

from itertools import permutations

def solution(n, weak, dist):
    length = len(weak)
    for i in range(length):
        weak.append(weak[i] + n) #1, 3, 4, 9, 10, 13, 15, 16, 21, 22
    answer = len(dist) + 1 #최솟값
    
    for start in range(length):
        for friends in list(permutations(dist, len(dist))):
            count = 1 #투입할 친구 수
            #해당 친구가 점검할 수 있는 마지막 위치
            position = weak[start] + friends[count - 1]
            
            #시작점부터 모든 취약 지점 확인
            for index in range(start, start + length):
                #점검할 수 있는 위치를 벗어나는 경우
                if position < weak[index]:
                    count += 1 #친구 더 투입
                    if count > len(dist): #더 투입 불가능하면 종료
                        break
                    position = weak[index] + friends[count - 1]
            
            answer = min(answer, count) 
    if answer >len(dist):
        return -1
    
    else:
        return answer

2. C++

#include <bits/stdc++.h>

using namespace std;

int solution(int n, vector<int> weak, vector<int> dist) {
    // 길이를 2배로 늘려서 '원형'을 일자 형태로 변경
    int length = weak.size();
    for (int i = 0; i < length; i++) {
        weak.push_back(weak[i] + n);
    }
    // 투입할 친구 수의 최솟값을 찾아야 하므로 len(dist) + 1로 초기화
    int answer = dist.size() + 1;
    // 0부터 length - 1까지의 위치를 각각 시작점으로 설정
    for (int start = 0; start < length; start++) {
        // 친구를 나열하는 모든 경우 각각에 대하여 확인
        do {
            int cnt = 1; // 투입할 친구의 수
            // 해당 친구가 점검할 수 있는 마지막 위치
            int position = weak[start] + dist[cnt - 1];
            // 시작점부터 모든 취약한 지점을 확인
            for (int index = start; index < start + length; index++) {
                // 점검할 수 있는 위치를 벗어나는 경우
                if (position < weak[index]) {
                    cnt += 1; // 새로운 친구를 투입
                    if (cnt > dist.size()) { // 더 투입이 불가능하다면 종료
                        break;
                    }
                    position = weak[index] + dist[cnt - 1];
                }
            }
            answer = min(answer, cnt); // 최솟값 계산
        } while(next_permutation(dist.begin(), dist.end()));
    }
    if (answer > dist.size()) {
        return -1;
    }
    return answer;
}