[프로그래머스 / Level 1] 체육복

https://programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복

문제 접근법

• 전체 인원이 체육복을 가지고 있다고 가정한다.
• reserve에 해당하는 인원들에게는 체육복을 하나씩 더 추가한다.
• lost에 해당하는 인원들에게는 체육복을 하나씩 제거한다.
• 자기의 체육복이 2개 일 때 체육복을 분배한다.

아래는 해당 코드입니다.

#include <string>
#include <vector>
 
using namespace std;
 
int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    int len = reserve.size();
    int lLen = lost.size();
    vector<int> v(n);
    
    for(int i = 0; i < n; i++)
        v[i] = 1;
    
    for(int i = 0; i < len; i++)
    {
        int temp = reserve[i];
        v[temp - 1]++;
    }
    
    for(int i = 0; i < lLen; i++)
    {
        int temp = lost[i];
        v[temp - 1]--;
    }
    
    for(int i = 0; i < n; i++)
    {
        if(v[i] == 2 && v[i - 1] == 0 && i > 0 )
        {
            v[i]--;
            v[i - 1]++;
        }
        
        if(v[i] == 2 && v[i + 1] == 0 && i < n - 1)
        {
            v[i]--;
            v[i + 1]++;
        }
    }
    
    for(int i = 0; i < n; i++)
    {
        if(v[i] >= 1)
            answer++;
    }
    
    return answer;
}