프로그래머스 Level 1 - 체육복 C++

코딩테스트 연습 - 체육복

#include <string>
#include <vector>

using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    int arr[31];
    
    for(int i = 0; i < n; i++)
        arr[i] = 1;
    
    for(int i = 0; i < lost.size(); i++)
        arr[lost[i]-1]--;
    
    for(int i = 0; i < reserve.size(); i++)
        arr[reserve[i]-1]++;
    
    for(int i = 0; i < n; i++){
        if(arr[i] == 2 && i > 0 && arr[i-1] == 0){
            arr[i]--;
            arr[i-1]++;
        }
        if(arr[i] == 2 && i < n && arr[i+1] == 0){
            arr[i]--;
            arr[i+1]++;
        }
    }
    
    for(int i = 0; i < n; i++)
        if(arr[i] >= 1)
                answer++;
    
    return answer;
}

---

 학생들의 수가 적어 무대포 방식으로 해도 작동이 된다.