level 1 - 체육복

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

코딩테스트 연습 - 체육복

import java.util.Arrays;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        
        int[] cloths = new int[n];
        
        for (int lostNum : lost) {
            cloths[lostNum - 1]--;
        }
        
        for (int reserveNum : reserve) {
            cloths[reserveNum - 1]++;
        }

        for (int i = 0; i < n; i++) {
            // 여분의 체육복이 있는 지 확인한다.
            if (cloths[i] <= 0) {
                continue;
            }

            // 이전 사람이 체육복을 가지지 않았다면 나눠준다.
            if (i - 1 >= 0 && cloths[i - 1] < 0) {
                cloths[i]--;
                cloths[i - 1]++;
            }
            // 앞 사람이 체육복을 가지지 않았다면 나눠준다.
            else if (i + 1 < n && cloths[i + 1] < 0) {
                cloths[i]--;
                cloths[i + 1]++;
            }
        }
        
        // 체육복을 가진 사람의 수를 계산해서 리턴한다.
        return (int) Arrays.stream(cloths).filter(i -> i >= 0).count();
    }
}