[C++] 프로그래머스 : 체육복

 

#include <string>
#include <vector>


using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {
    
    vector<int> v(n+1, 1);
    
    
    for(int i = 0; i<lost.size(); i++){
        v[lost[i]] -= 1;
    }
    for(int i = 0; i<reserve.size(); i++){
        v[reserve[i]] += 1;
    }
    
    
    for(int i = 1; i<v.size(); i++){
        if(v[i] == 0){
            if(v[i-1] == 2){
                v[i-1] -= 1;
                v[i] += 1;
            }
            else if(v[i+1] == 2){
                v[i+1] -= 1;
                v[i] += 1;
            }
        }
    }

    int answer = 0;
    
    for(int i = 1; i < v.size(); i++){
        if(v[i] != 0){
            answer++;
        }
    }
    
    return answer;
}