[programmers] 체육복 / Java

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

코딩테스트 연습 - 체육복

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int answer = 0;
        
        // -1: 도난
	// 0 : 도난 x, 여벌 x (도난당했는데 여벌 있는 경우 포함)
	// 1 : 도난 x, 여벌 o
        int all[] = new int[n];

	// 도난당한 사람 체크
        for (int i : lost) {
            all[i - 1]--;
        }
        
        // 여벌 있는 사람 체크
        for (int i : reserve) {
            all[i - 1]++;
        }

        for (int i = 0; i < all.length; i++) {
            if (all[i] < 0) { // 도난당했을 경우 옆에서 줄 수 있는지 확인
                if ((i != 0) && all[i - 1] > 0) {
                    all[i]++;
                    all[i - 1]--;
                } else if ((i + 1 != all.length) && all[i + 1] > 0) {
                    all[i]++;
                    all[i + 1]--;
                }
            }
        }

        for (int i : all) {
            if (i >= 0) {
                answer++;
            }
        }
        return answer;
    }
}