프로그래머스 - 체육복

 코딩테스트 고득점 Kit 탐욕법(Greedy) Level 1 체육복 문제이다. 이 문제는 기초적인 그리디 알고리즘 문제로, 문제에서 제시한 그대로 구현하면 정답이 나온다.

 

#include <string>
#include <vector>

using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    vector<int> students(n + 1, 1);
    
    for (auto i : lost)
    {
        students[i]--;
    }
    
    for (auto i : reserve)
    {
        students[i]++;
    }
    
    for (int i = 1; i <= n; i++)
    {
        if (i == 1)
        {
            if (students[i] == 2 && students[i + 1] == 0)
            {
                students[i]--;
                students[i + 1]++;
            }
            continue;
        }
        if (i == n)
        {
            if (students[i] == 2 && students[i - 1] == 0)
            {
                students[i]--;
                students[i - 1]++;
            }
            continue;
        }
        
        if (students[i] == 2 && students[i - 1] == 0)
        {
            students[i]--;
            students[i - 1]++;
        }
        
        if (students[i] == 2 && students[i + 1] == 0)
        {
            students[i]--;
            students[i + 1]++;
        }
    }
    
    for (int i = 1; i <= n; i++)
    {
        if (students[i] == 1 || students[i] == 2)
        {
            answer++;
        }
    }
    
    return answer;
}