[프로그래머스] 체육복 - Java

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

코딩테스트 연습 - 체육복

먼저, 각 학생별 체육복 도난 및 여분 보유 여부를 정리한다.

• 각 학생들의 체육복 보유 수량을 나타내는 정수 배열을 선언하고 그 값을 모두 1로 초기화한다. (시작시점에는 모든 학생이 체육복을 가지고 있다.)
• 도둑질 당한 학생들의 번호에 해당하는 원소들에서 1씩 뺀다.
• 여분 체육복을 가지고 온 학생들의 번호에 해당하는 원소들에 1씩 더한다.

이후, 맨 앞 학생부터 차례대로 체육복 갯수가 0개인지 확인한다.

• 해당 학생이 보유한 체육복의 갯수가 0개일 경우
  • 앞 학생이 체육복을 2개 가지고 있을 경우 빌린다. (현재학생++, 앞 학생--)
  • 앞 학생에게서 빌리지 못했고, 뒷 학생이 체육복을 2개 가지고 있는 경우 체육복을 빌린다. (현재학생++, 뒷 학생--)

최종적으로 배열에서 0 이상의 값을 가진 원소의 갯수를 세어 반환한다.

 

문제에서는 여분의 체육복만 빌려줄 수 있도록 조건이 지정되어 있는데, 내 체육복을 빌려주고 나는 다른 학생의 체육복을 빌려입을 수 있도록 조건이 설정되어 있었다면 좀 더 재밌었을 듯 하다.

 

소스코드

import java.util.Arrays;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int[] students = new int[n + 1];
        Arrays.fill(students, 1);
        students[0] = 0; // dummy data
        // 도둑질 결과
        for (int studentNumber : lost) {
            students[studentNumber]--;
        }
        // 여분 체육복 확인
        for (int studentNumber : reserve) {
            students[studentNumber]++;
        }

        // 첫 번째 학생
        if (students[1] == 0 && students[2] > 1) {
            students[1]++;
            students[2]--;
        }
        // 중간 학생들
        for (int i = 2; i < n; i++) {
            // i번째 학생이 체육본 도난당했을 경우
            if (students[i] == 0) {
                if (students[i - 1] > 1) {
                    // 앞 학생 확인
                    students[i - 1]--;
                    students[i]++;
                } else if (students[i + 1] > 1) {
                    // 뒤 학생 확인
                    students[i + 1]--;
                    students[i]++;
                }
            }
        }
        // 마지막 학생
        if (students[n] == 0 && students[n - 1] > 1) {
            students[n]++;
            students[n - 1]--;
        }

        // count
        int count = 0;
        for (int student : students) {
            if (student > 0) {
                count++;
            }
        }
        return count;
    }
}