프로그래머스-Lv1 42862. 체육복

[devLupin]

문제 링크

내 코드

[깔끔한 풀이]

• 나도 배열을 사용해서 체크했는데, 조금 지저분함.
• 1, 0, -1로 분기를 체크한 게 인상 깊었음.
• 나는 정렬을 했는데 배열을 통해 전부 탐색하면 정렬도 필요없음.

#include <string>
#include <vector>

using namespace std;
int student[35];
int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    for(int i : reserve) student[i] += 1;
    for(int i : lost) student[i] += -1;
    for(int i = 1; i <= n; i++) {
        if(student[i] == -1) {
            if(student[i-1] == 1) 
                student[i-1] = student[i] = 0;
            else if(student[i+1] == 1) 
                student[i] = student[i+1] = 0;
        }
    }
    for(int i  = 1; i <=n; i++)
        if(student[i] != -1) answer++;

    return answer;
}