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

문제링크: programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복

 

풀이

 

학생이 가지고 있는 체육복의 수를 저장하는 배열 student를 만들고, 체육복을 잃어버린 학생은 -1, 여분의 체육복이 있는 학생은 1을 더해줍니다. 1번부터 n번 학생 중에서 체육복이 없는 학생의 경우 먼저 바로 앞에 학생(student[i - 1])에게 체육복을 빌릴 수 있는지 확인합니다. 체육복을 빌릴 수 없다면 뒤에 학생(student[i + 1])에게 빌릴 수 있는지 확인합니다. 

 

최종적으로 student 배열에 -1이 저장된 학생은 체육수업에 참가하지 못하므로 answer에서 제외합니다. 

 

이때, 학생 번호를 그대로 배열의 index로 사용하는 것에 주의합니다. 만약 5명의 학생이 있다면 각 학생의 체육복 개수는 student[1], student[2], student[3], student[4], student[5]에 저장됩니다. student[i - 1]과 student[i + 1]에 접근해야 하므로 배열의 크기는 최대 학생 수에서 2 이상 크게 잡아야 합니다. 

 

 

 

소스코드

 

#include <string>
#include <vector>
 
using namespace std;
int student[33];
int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    
    for(int i: lost) student[i] -= 1;
    for(int i: reserve) student[i] += 1;
 
    for(int i = 1; i <= n; i++) {
        if(student[i] == -1) {
            if(student[i - 1] == 1) {
                student[i] = student[i - 1] = 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;
}

 

테스트 결과

 

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)