프로그래머스 체육복

문제

programmers.co.kr/learn/courses/30/lessons/42862?language=java

코딩테스트 연습 - 체육복

코드

프로그래머스 체육복

class Solution {
       
    public int solution(int n, int[] lost, int[] reserve) {

        //students
        int[] students = new int[n + 1];    //0번 인덱스 사용 X

        //lost check
        for (int i = 0; i < lost.length; i++)
            students[lost[i]] = -1;

        //reserve check
        for (int i = 0; i < reserve.length; i++)
            students[reserve[i]]++;         //lost && reserve 대비

        //borrow
        int idx = -1;
        for (int i = 0; i < lost.length; i++) {
            idx = lost[i];  //lost 학생 번호
            if (students[idx] != -1) continue;   //lost && reserve 인 경우 continue
            //다음사람에게 빌리기
            if (idx < n && students[idx + 1] == 1) {    //인덱스 체크 && ~~~
                students[idx + 1] = 0;
                students[idx] = 0;
                continue;
            }
            //이전사람에게 빌리기
            if (idx - 1 >= 1 && students[idx - 1] == 1) {       //인덱스 체크 && ~~~
                students[idx - 1] = 0;
                students[idx] = 0;
            }
        }

        //체육복 없는 학생 카운트
        int cnt = 0;
        for (int i = 1; i < students.length; i++) {
            if (students[i] == -1) cnt++;
        }

        return n - cnt;
    }//method
    
}