[프로그래머스 42862] 체육복

문제링크

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

코딩테스트 연습 - 체육복

 

문제난이도

Level 1

 

문제풀이

1. 옷을 잃어버리 않은 학생들은 수업에 참여 할 수 있다

2. 옷을 잃어버렸지만 여벌을 갖고 있는 학생들은 수업에 참여가 가능하다

   ★주의★

    - 옷을 잃어버렸지만 여벌을 갖고 있는 학생은 다른 학생에게 옷을 빌려줄 수 없다

3. 여벌옷을 갖고있는 남은 학생들 중 바로 앞 또는 뒤의 번호인 경우 체육볼을 빌려 수업에 참가한다

 

 

자세한 풀이는 소스코드를 통해 확인해보자

소스코드

 

int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = n - lost.size();    //기존에 옷을 갖고 있는 학생수
 
    for (int i = 0; i < lost.size(); i++) {
        for (int j = 0; j < reserve.size(); j++) {
            //체육복을 잃어버린 학생중 여벌을 갖고 있는 학생 체크
            if (lost[i] == reserve[j]) {
                answer++;
                reserve[j] = -1;
                lost[i] = -1;
                break;
            }
        }
    }
 
    for (int i = 0; i < lost.size(); i++) {
        if (lost[i] == -1) {
            continue;
        }
        
        for (int j = 0; j < reserve.size(); j++) {
            //여벌 옷을 갖고 있는 학생 중, 옷을 잃어버린 학생의 번호의 -1 또는 +1 번호인 경우
            if (lost[i] - 1 == reserve[j] || lost[i] + 1 == reserve[j]) {
                answer++;
                reserve[j] = -1;
                break;
            }
        }
    }
 
    return answer;
}